First commit

pyproject.toml

@@ -17,10 +17,10 @@ packages = [{ include = "qiboml", from = "src" }]
 python = ">=3.9,<3.13"
 numpy = "^1.26.4"
 keras = { version = "^3.0.0", optional = true }
-tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'", optional = true }
+tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'"}
 # 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 }
+torch = { version = "^2.3.1"}
 qibo = {git="https://github.com/qiboteam/qibo"}
 jax = "^0.4.25"
 jaxlib = "^0.4.25"
@@ -46,11 +46,10 @@ pdbpp = "^0.10.3"
 optional = true
 
 [tool.poetry.group.tests.dependencies]
-torch = "^2.3.1"
-tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux'" }
 pytest = "^7.2.1"
 pylint = "3.1.0"
 pytest-cov = "4.0.0"
+qibojit = "^0.1.7"
 
 [tool.poetry.group.benchmark.dependencies]
 pytest-benchmark = { version = "^4.0.0", extras = ["histogram"] }

src/qiboml/__init__.py

@@ -12,7 +12,7 @@
 try:
     from tensorflow import Tensor as tf_tensor
 
-    from qiboml.models import keras
+    from qiboml.interfaces import keras
 
     ndarray = Union[ndarray, tf_tensor]
 except ImportError:  # pragma: no cover
@@ -21,7 +21,7 @@
 try:
     from torch import Tensor as pt_tensor
 
-    from qiboml.models import pytorch
+    from qiboml.interfaces import pytorch
 
     ndarray = Union[ndarray, pt_tensor]
 except ImportError:  # pragma: no cover