Spaces:
Runtime error
Runtime error
Update scheduler.py
Browse files- scheduler.py +4 -3
scheduler.py
CHANGED
@@ -10,8 +10,9 @@ class ZipScheduler(CommitScheduler):
|
|
10 |
with self.lock:
|
11 |
if not any(self.folder_path.iterdir()):
|
12 |
return
|
13 |
-
archive_file = tempfile.NamedTemporaryFile()
|
14 |
-
|
|
|
15 |
shutil.rmtree(self.folder_path, ignore_errors=True)
|
16 |
self.folder_path.mkdir(parents=True, exist_ok=True)
|
17 |
|
@@ -20,7 +21,7 @@ class ZipScheduler(CommitScheduler):
|
|
20 |
repo_type=self.repo_type,
|
21 |
revision=self.revision,
|
22 |
path_in_repo=f"{uuid.uuid4()}.zip",
|
23 |
-
path_or_fileobj=archive_file.name
|
24 |
)
|
25 |
|
26 |
archive_file.close()
|
|
|
10 |
with self.lock:
|
11 |
if not any(self.folder_path.iterdir()):
|
12 |
return
|
13 |
+
archive_file = tempfile.NamedTemporaryFile(suffix=".zip")
|
14 |
+
archive_name = archive_file.name.split(".")[0] # `make_archive` automatically append `.zip`
|
15 |
+
shutil.make_archive(base_name=archive_name, format='zip', root_dir=self.folder_path)
|
16 |
shutil.rmtree(self.folder_path, ignore_errors=True)
|
17 |
self.folder_path.mkdir(parents=True, exist_ok=True)
|
18 |
|
|
|
21 |
repo_type=self.repo_type,
|
22 |
revision=self.revision,
|
23 |
path_in_repo=f"{uuid.uuid4()}.zip",
|
24 |
+
path_or_fileobj=archive_file.name,
|
25 |
)
|
26 |
|
27 |
archive_file.close()
|