utilities

Lightning-Universe · May 5, 2023 · 6b51d57 · 6b51d57
1 parent 4bb0d41
commit 6b51d57
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -12,4 +12,4 @@ jsonargparse[signatures] >=3.17.0, <=4.9.0
 click >=7.1.2, <=8.1.3
 protobuf <=3.20.1
 fsspec[http] >=2022.5.0,<=2022.7.1
-lightning-utilities >=0.4.1
+lightning-utilities >=0.3.0
diff --git a/src/flash/core/data/base_viz.py b/src/flash/core/data/base_viz.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 from typing import Any, Dict, List, Set, Tuple
 
-from lightning_utilities.core.overrides import is_overridden
+from pytorch_lightning.utilities.model_helpers import is_overridden
 
 from flash.core.data.callback import BaseDataFetcher
 from flash.core.data.utils import _CALLBACK_FUNCS