possible fix for mac error (#200)

brainglobe · Jul 14, 2023 · 224e0b3 · 224e0b3
1 parent b387053
commit 224e0b3
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/cellfinder_core/__init__.py b/src/cellfinder_core/__init__.py
@@ -10,12 +10,15 @@
 try:
     TF_VERSION = version("tensorflow")
 except PackageNotFoundError as e:
-    raise PackageNotFoundError(
-        f"cellfinder tools cannot be invoked without tensorflow. "
-        f"Please install tensorflow into your environment to use cellfinder tools. "
-        f"For more information, please see "
-        f"https://github.com/brainglobe/brainglobe-meta#readme."
-    ) from e
+    try:
+        TF_VERSION = version("tensorflow-macos")
+    except PackageNotFoundError as e:
+        raise PackageNotFoundError(
+            f"cellfinder tools cannot be invoked without tensorflow. "
+            f"Please install tensorflow into your environment to use cellfinder tools. "
+            f"For more information, please see "
+            f"https://github.com/brainglobe/brainglobe-meta#readme."
+        ) from e
 
 __author__ = "Adam Tyson, Christian Niedworok, Charly Rousseau"
 __license__ = "BSD-3-Clause"