fix importguard

pytorch · Dec 12, 2024 · 6f64f52 · 6f64f52
1 parent b9f7303
commit 6f64f52
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torchtune/utils/_import_guard.py b/torchtune/utils/_import_guard.py
@@ -14,7 +14,10 @@
 )
 
 _TORCHDATA_MIN_VERSION = "0.10.0"
-if importlib.util.find_spec("torchdata.nodes") is not None:
+if (
+    importlib.util.find_spec("torchdata") is not None
+    and importlib.util.find_spec("torchdata.nodes") is not None
+):
     _TORCHDATA_INSTALLED = True
 else:
     _TORCHDATA_INSTALLED = False