build: made tf optional

qiboteam · Jul 22, 2024 · 2ac5168 · 2ac5168
1 parent c78f189
commit 2ac5168
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ packages = [{ include = "qiboml", from = "src" }]
 python = ">=3.9,<3.12"
 numpy = "^1.26.4"
 numba = "^0.59.0"
-tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'" }
+tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'", optional = true }
 # TODO: the marker is a temporary solution due to the lack of the tensorflow-io 0.32.0's wheels for Windows, this package is one of
 # the tensorflow requirements
 torch = { version = "^2.3.1", optional = true }

diff --git a/src/qiboml/models/keras.py b/src/qiboml/models/keras.py
@@ -15,7 +15,6 @@
 # import tensorflow as tf
 
 
-
 """
 def _keras_factory(module):
     for name, layer in inspect.getmembers(module, inspect.isclass):
-Original file line number
+Diff line change
@@ Expand Up / @@ -15,7 +15,6 @@ @@
     # import tensorflow as tf
     """
     def _keras_factory(module):
         for name, layer in inspect.getmembers(module, inspect.isclass):
@@ Expand Down @@