fix failing dataset unit tests for py38 (#2263)

unit8co · Mar 4, 2024 · 4de71e4 · 4de71e4
1 parent 2117bf3
commit 4de71e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/darts/tests/datasets/test_datasets.py b/darts/tests/datasets/test_datasets.py
@@ -507,7 +507,7 @@ def test_inference_dataset_output_chunk_shift(self, config):
             target = self.target1[: -(ocl + ocs)]
 
             ds_covs = {}
-            ds_init_params = set(inspect.signature(ds_cls).parameters)
+            ds_init_params = set(inspect.signature(ds_cls.__init__).parameters)
             for cov_type in ["covariates", "past_covariates", "future_covariates"]:
                 if cov_type in ds_init_params:
                     ds_covs[cov_type] = self.cov1
@@ -1388,7 +1388,7 @@ def test_sequential_training_dataset_output_chunk_shift(self, config):
             target = self.target1[: -(ocl + ocs)]
 
             ds_covs = {}
-            ds_init_params = set(inspect.signature(ds_cls).parameters)
+            ds_init_params = set(inspect.signature(ds_cls.__init__).parameters)
             for cov_type in ["covariates", "past_covariates", "future_covariates"]:
                 if cov_type in ds_init_params:
                     ds_covs[cov_type] = self.cov1