cached_property is available in functools for Python 3.8+

strongio · Apr 25, 2024 · 898bf65 · 898bf65
1 parent 3256cb1
commit 898bf65
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,6 @@
     packages=find_packages(include='torchcast.*'),
     zip_safe=False,
     install_requires=[
-        'backports.cached-property',
         'torch>=1.12',
         'numpy>=1.4',
     ],

diff --git a/torchcast/state_space/predictions.py b/torchcast/state_space/predictions.py
@@ -1,14 +1,11 @@
+from functools import cached_property
 from typing import Tuple, Union, Optional, Dict, Iterator, Sequence
 
+import numpy as np
 import torch
 from torch import nn, Tensor
 
-import numpy as np
-
-from backports.cached_property import cached_property
-
 from torchcast.internals.utils import get_nan_groups, is_near_zero
-
 from torchcast.utils.data import TimeSeriesDataset
 from torchcast.utils.outliers import mahalanobis_dist