ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-01 17:43:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix fastai version check import

Browse Source
This commit is contained in:
allegroai 2022-02-19 23:05:55 +02:00
parent 73b64f6ca7
commit 3a3c947bf1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/binding/frameworks/fastai_bind.py
View File

@ -2,7 +2,7 @@ import sys
import numpy as np import numpy as np
from packaging import version from clearml.utilities.version import Version
from . import _patched_call from . import _patched_call
from .tensorflow_bind import WeightsGradientHistHelper from .tensorflow_bind import WeightsGradientHistHelper
@ -23,7 +23,7 @@ class PatchFastai(object):
# noinspection PyBroadException # noinspection PyBroadException
try: try:
if version.parse(fastai.__version__) < version.parse("2.0.0"): if Version(fastai.__version__) < Version("2.0.0"):
PatchFastaiV1.update_current_task(task) PatchFastaiV1.update_current_task(task)
PatchFastaiV1.patch_model_callback() PatchFastaiV1.patch_model_callback()
PostImportHookPatching.add_on_import("fastai", PatchFastaiV1.patch_model_callback) PostImportHookPatching.add_on_import("fastai", PatchFastaiV1.patch_model_callback)