Remove cached model directories recursively

2026-10-27 16:52:33 +00:00 · 2026-10-27 16:52:33 +00:00 · 6c4b33a418
commit 6c4b33a418
parent 4f603817b2
1 changed files with 27 additions and 2 deletions
--- a/great_ai/large_file/large_file/large_file_base.py
+++ b/great_ai/large_file/large_file/large_file_base.py
@ -332,7 +332,10 @@ class LargeFileBase(ABC):
            logger.info(
                f"Deleting file from cache to meet quota (max_cache_size={self.max_cache_size}): {file}"
            )
-            os.unlink(file)
+            if file.is_dir():
+                shutil.rmtree(file)
+            else:
+                file.unlink()

    @abstractmethod
    def _find_remote_instances(self) -> List[DataInstance]: