@@ -55,7 +55,7 @@ class TestDetectionDatasetCaching(unittest.TestCase):
 
         datasets = [
             DummyDetectionDataset(input_dim=(640, 512), ignore_empty_annotations=False, class_inclusion_list=class_inclusion_list,
-                                  cache=True, cache_path=self.temp_cache_dir, data_dir='/home/')
+                                  cache=True, cache_dir=self.temp_cache_dir, data_dir='/home/')
             for class_inclusion_list in [["class_0", "class_1", "class_2"], ["class_0"], ["class_1"], ["class_2"], ["class_1", "class_2"]]
         ]
 
@@ -70,7 +70,7 @@ class TestDetectionDatasetCaching(unittest.TestCase):
 
         datasets = [
             DummyDetectionDataset(input_dim=(640, 512), ignore_empty_annotations=True, class_inclusion_list=class_inclusion_list,
-                                  cache=True, cache_path=self.temp_cache_dir, data_dir='/home/')
+                                  cache=True, cache_dir=self.temp_cache_dir, data_dir='/home/')
             for class_inclusion_list in [["class_0", "class_1", "class_2"], ["class_0"], ["class_1"], ["class_2"], ["class_1", "class_2"]]
         ]
 
@@ -87,12 +87,12 @@ class TestDetectionDatasetCaching(unittest.TestCase):
         self.assertEqual(0, self._count_cached_array())
 
         _ = DummyDetectionDataset(input_dim=(640, 512), ignore_empty_annotations=True,
-                                  cache=True, cache_path=self.temp_cache_dir, data_dir='/home/')
+                                  cache=True, cache_dir=self.temp_cache_dir, data_dir='/home/')
         self.assertEqual(1, self._count_cached_array())
 
         for _ in range(5):
             _ = DummyDetectionDataset(input_dim=(640, 512), ignore_empty_annotations=True,
-                                      cache=True, cache_path=self.temp_cache_dir, data_dir='/home/')
+                                      cache=True, cache_dir=self.temp_cache_dir, data_dir='/home/')
             self.assertEqual(1, self._count_cached_array())
 
         self._empty_cache()