fix CI

sustainability-lab · Oct 27, 2023 · 8dc3105 · 8dc3105
1 parent 1a0bc77
commit 8dc3105
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -9,7 +9,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.9, 3.10, 3.11]
+        python-version: ["3.9", "3.10", "3.11"]
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}

diff --git a/README.md b/README.md
@@ -42,6 +42,6 @@ print(result.keys()) # dict_keys(['epoch_losses', 'iter_losses'])
 ## Adhoc
 ### Count number of parameters in a model
 ```python
-from susml.torch.utils import count_parameters
+from susml.torch.utils import count_params
 n_params = count_params(mlp)
 ```