Revert "minor fix"

This reverts commit c95d386.
awslabs · Oct 17, 2024 · 52d85a9 · 52d85a9
1 parent c95d386
commit 52d85a9
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -20,10 +20,10 @@ jobs:
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
 
-    - name: Set up Python
+    - name: Set up Python 3.12
       uses: actions/setup-python@v4
       with:
-        python-version: 3.11
+        python-version: 3.12
 
     - name: Install dependencies
       run: |

diff --git a/.github/workflows/doctests.yaml b/.github/workflows/doctests.yaml
@@ -8,7 +8,7 @@ jobs:
       max-parallel: 4
       fail-fast: false
       matrix:
-        python-version: ['3.11']
+        python-version: ['3.12']
         platform: [ubuntu-latest]
 
     runs-on: ${{ matrix.platform }}

diff --git a/.github/workflows/style_type_checks.yml b/.github/workflows/style_type_checks.yml
@@ -11,7 +11,7 @@ jobs:
     - uses: extractions/setup-just@v1
     - uses: actions/setup-python@v4
       with:
-          python-version: '3.11'
+          python-version: '3.12'
     - name: Install dependencies
       run: |
         pip install .

diff --git a/test/torch/distribution/test_negative_binomial.py b/test/torch/distribution/test_negative_binomial.py
@@ -49,7 +49,7 @@ def test_custom_neg_bin_cdf(total_count, probs, value):
 @pytest.mark.parametrize("probs", [0.1, 0.5, 0.8])
 @pytest.mark.parametrize("total_count", [3, 7, 100])
 @pytest.mark.parametrize("value", [0.1, 0.5, 0.9])
-def test_custom_neg_bin_icdf(total_count, probs, value):
+def test_custom_studentt_icdf(total_count, probs, value):
     torch_dist = NegativeBinomial(total_count=total_count, probs=probs)
     scipy_dist = torch_dist.scipy_nbinom