Skip to content

Commit

Permalink
Merge branch 'main' of github.com:artefactory/choice-learn-private
Browse files Browse the repository at this point in the history
  • Loading branch information
VincentAuriau committed Dec 23, 2024
2 parents 3ff9d54 + d035e15 commit 1597ca3
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 2 deletions.
6 changes: 6 additions & 0 deletions .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,13 @@ jobs:
- name: Build coverage file
run: |
set -e
echo before: $?
python -m pytest tests/
echo exit_code btw: $?
pytest -n auto --junitxml=pytest.xml --cov-report=term-missing --cov=choice_learn tests/ | tee pytest-coverage.txt
echo exit_code: $?
cat pytest-coverage.txt
- name: Pytest coverage comment
uses: VincentAuriau/pytest-coverage-comment@main
Expand Down
2 changes: 1 addition & 1 deletion choice_learn/models/latent_class_base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
numpy>=1.24
pandas>=1.5
tensorflow>=2.14, <2.17
tensorflow>=2.18, <2.19
tensorflow_probability>=0.22
tf_keras<3
tqdm>=4.0

0 comments on commit 1597ca3

Please sign in to comment.