fix core50 test to avoid out of memory issues
""" CORe50 Tests"""
import unittest
+import os
+
from avalanche.benchmarks.classic import CORe50
@@ -19,17 +21,18 @@ from avalanche.benchmarks.classic import CORe50
class CORe50Test(unittest.TestCase):
def test_core50_ni_scenario(self):
- # for now we disable it as it takes a while to download the CORe50
- # dataset.
+ if "FAST_TEST" in os.environ:
+ pass
+ else:
+ scenario = CORe50(scenario="ni")
+ for task_info in scenario:
pass
-
- # scenario = CORe50(scenario="ni")
- # for task_info in scenario:
- # self.assertIsInstance(task_info, IExperience)
def test_core50_nc_scenario(self):
+ if "FAST_TEST" in os.environ:
+ pass
+ else:
benchmark_instance = CORe50(scenario='nc')
-
self.assertEqual(1, len(benchmark_instance.test_stream))
classes_in_test = benchmark_instance.\
