ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make sure local packages with multi-files are marked as "package"

Browse Source
This commit is contained in:
allegroai 2020-10-12 10:50:46 +03:00
parent 1e4ab0510c
commit 0b5c22d6dd
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
trains/binding/frameworks/__init__.py
View File

@ -429,7 +429,8 @@ class WeightsFileHandler(object):
weights_filename=temp_file, auto_delete_file=True, target_filename=target_filename, weights_filename=temp_file, auto_delete_file=True, target_filename=target_filename,
update_comment=False) update_comment=False)
else: else:
trains_out_model.update_weights(weights_filename=None, register_uri=model_info.local_model_path) trains_out_model.update_weights(
weights_filename=None, register_uri=model_info.local_model_path, is_package=bool(len(files) > 1))
# update back the internal Model lookup, and replace the local file with our file # update back the internal Model lookup, and replace the local file with our file
# noinspection PyProtectedMember # noinspection PyProtectedMember