np.product -> np.prod

artefactory · Dec 23, 2024 · 2e48dcb · 2e48dcb
1 parent 5977dbd
commit 2e48dcb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -28,7 +28,7 @@ jobs:
 
       - name: Build coverage file
         run: |
-          pytest --junitxml=pytest.xml --cov-report=term-missing --cov=choice_learn tests/ | tee pytest-coverage.txt
+          pytest -n auto --junitxml=pytest.xml --cov-report=term-missing --cov=choice_learn tests/ | tee pytest-coverage.txt
           cat pytest-coverage.txt
 
       - name: Pytest coverage comment

diff --git a/choice_learn/models/latent_class_base_model.py b/choice_learn/models/latent_class_base_model.py
@@ -360,7 +360,7 @@ def _lbfgs_train_step(self, choice_dataset, sample_weight=None):
         part = []  # partition indices
 
         for i, shape in enumerate(shapes):
-            n = np.product(shape)
+            n = np.prod(shape)
             idx.append(tf.reshape(tf.range(count, count + n, dtype=tf.int32), shape))
             part.extend([i] * n)
             count += n