diff --git a/src/cellfinder_core/__init__.py b/src/cellfinder_core/__init__.py index e0b2b8e7..746b2d2b 100644 --- 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"