Wauplin HF Staff commited on
Commit
a4df400
·
1 Parent(s): 0cc5070

Update scheduler.py

Browse files
Files changed (1) hide show
  1. 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
- shutil.make_archive(base_name=archive_file.name, format='zip', root_dir=self.folder_path)
 
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 + ".zip",
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()