integration cache test fix solved (#935)

aimclub · Oct 18, 2022 · f0c69cc · f0c69cc
1 parent 023cbeb
commit f0c69cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fedot/core/caching/base_cache_db.py b/fedot/core/caching/base_cache_db.py
@@ -86,7 +86,7 @@ def _del_prev_temps(self):
         """
         Deletes previously generated unused DB files.
         """
-        for file in self.db_path.parent.glob(f'cache_*{self._db_suffix}'):
+        for file in self.db_path.glob(f'cache_*{self._db_suffix}'):
             try:
                 pid = int(file.stem.split('_')[-1])
             except ValueError:

diff --git a/test/integration/cache/test_parallel_cache_files.py b/test/integration/cache/test_parallel_cache_files.py
@@ -39,7 +39,7 @@ def test_parallel_cache_files():
     cpus = multiprocessing.cpu_count()
     try:
         with multiprocessing.Pool(processes=cpus) as pool:
-            pool.map(run_example, tasks)
+            list(pool.imap(run_example, tasks))
     except sqlite3.OperationalError:
         assert False, 'DBs collides'