check codecov secret

Author: givasile

effector/models.py

@@ -125,7 +125,7 @@ def predict(self, x: np.ndarray) -> np.ndarray:
         Returns:
             Output of the model, shape (N,)
         """
-        y = np.exp(x[:, 3])
+        y = np.exp(x[:, 2])
 
         mask1 = (x[:, 1] < 0) & (x[:, 2] < 0)
         mask2 = (x[:, 1] < 0) & (x[:, 2] >= 0)
@@ -150,7 +150,7 @@ def jacobian(self, x: np.ndarray) -> np.ndarray:
         """
         y = np.zeros(x.shape)
 
-        y[:, 3] = np.exp(x[:, 3])
+        y[:, 2] = np.exp(x[:, 2])
 
         mask1 = (x[:, 1] < 0) & (x[:, 2] < 0)
         mask2 = (x[:, 1] < 0) & (x[:, 2] >= 0)

tests/test_models.py

@@ -0,0 +1,19 @@
+import effector
+
+def test_models():
+    # Test ConditionalInteraction model
+    for model in [
+        effector.models.ConditionalInteraction(),
+        effector.models.DoubleConditionalInteraction(),
+        effector.models.ConditionalInteraction4Regions(),
+        effector.models.GeneralInteraction()
+    ]:
+        x = effector.datasets.IndependentUniform(dim=3, low=-1, high=1).generate_data(1000, seed=21)
+
+        # Test predict method
+        y_pred = model.predict(x)
+        assert y_pred.shape == (1000,)
+
+        # Test jacobian method
+        jacobian = model.jacobian(x)
+        assert jacobian.shape == (1000, 3)