ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix type-hints and docstrings

Browse Source
This commit is contained in:
allegroai
2020-06-01 11:00:42 +03:00
parent 43dac458df
commit 2066d9ff9d
4 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -57,7 +57,9 @@ class WeightsFileHandler(object):
@staticmethod
def add_pre_callback(callback_function):
# type: (Callable[[Union['load', 'save'], str, str, Task], str]) -> int
# type: (Callable[[str, str, str, Task], str]) -> int
# callback is Callable[[Union['load', 'save'], str, str, Task], str]
if callback_function in WeightsFileHandler._model_pre_callbacks.values():
return [k for k, v in WeightsFileHandler._model_pre_callbacks.items() if v == callback_function][0]
@@ -70,7 +72,9 @@ class WeightsFileHandler(object):
@staticmethod
def add_post_callback(callback_function):
# type: (Callable[[Union['load', 'save'], Model, str, str, str, Task], Model]) -> int
# type: (Callable[[str, Model, str, str, str, Task], Model]) -> int
# callback is Callable[[Union['load', 'save'], Model, str, str, str, Task], Model]
if callback_function in WeightsFileHandler._model_post_callbacks.values():
return [k for k, v in WeightsFileHandler._model_post_callbacks.items() if v == callback_function][0]