update dataset upload

deeppavlov · Nov 13, 2023 · e6986d5 · e6986d5
1 parent d66e71b
commit e6986d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dff/stats/cli.py b/dff/stats/cli.py
@@ -281,7 +281,7 @@ def make_zip_config(parsed_args: argparse.Namespace) -> Path:
         logger.info(f"Saving the archive to {outfile_name}.")
         shutil.make_archive(outfile_name[: outfile_name.rindex(".zip")], format="zip", root_dir=temp_config_dir)
         for path in Path(temp_config_dir).iterdir():
-            if path != database_dir and path != dataset_dir:
+            if path.stem != database_dir.stem and path.stem != dataset_dir.stem:
                 if path.is_dir():
                     shutil.rmtree(path)