From 3a3c947bf124517aea3d80ba512c8a54c5cccadb Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sat, 19 Feb 2022 23:05:55 +0200 Subject: [PATCH] Fix fastai version check import --- clearml/binding/frameworks/fastai_bind.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clearml/binding/frameworks/fastai_bind.py b/clearml/binding/frameworks/fastai_bind.py index 59353b63..ce1fb703 100644 --- a/clearml/binding/frameworks/fastai_bind.py +++ b/clearml/binding/frameworks/fastai_bind.py @@ -2,7 +2,7 @@ import sys import numpy as np -from packaging import version +from clearml.utilities.version import Version from . import _patched_call from .tensorflow_bind import WeightsGradientHistHelper @@ -23,7 +23,7 @@ class PatchFastai(object): # noinspection PyBroadException 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.patch_model_callback() PostImportHookPatching.add_on_import("fastai", PatchFastaiV1.patch_model_callback)