fixed dependencies; enabled tests

akhilpandey95 · Oct 19, 2023 · 481f30f · 481f30f
1 parent bcd072c
commit 481f30f
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 192 deletions.
diff --git a/.github/workflows/publish-to-pypi.yml b/.github/workflows/publish-to-pypi.yml
@@ -37,8 +37,8 @@ jobs:
       run: |
         python -m poetry install
 
-#    - name: Test
-#      run: poetry run python3 -m unittest discover
+    - name: Test
+      run: poetry run python3 -m unittest discover
 
     - name: Install pdoc3
       run: pip install pdoc3

diff --git a/README.md b/README.md
@@ -81,5 +81,8 @@ If you'd like to contribute, be sure to check out our [contributing guide](./CON
 Thanks goes to these incredible people.
 
 <a href="https://github.com/akhilpandey95/gnnNAS/graphs/contributors">
-  <img src="https://contrib.rocks/image?repo=akhilpandey95/gnnNAS" />
+  <img style="border-radius: 50%" src="https://contrib.rocks/image?repo=akhilpandey95/gnnNAS" />
+</a>
+<a href="https://github.com/akhilpandey95/gnnNAS/graphs/contributors">
+  <img style="border-radius: 50%" src="https://contrib.rocks/image?repo=harishsiravuri/kgforge" />
 </a>
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,16 +12,10 @@ numpy = "^1.26.1"
 scikit-learn = "^1.3.1"
 pdoc3 = "^0.10.0"
 pytest = "^7.4.2"
+torch = "1.12.1"
 torch-geometric = "^2.4.0"
-torch = "^2.1.0+cu118"
 
 
-[[tool.poetry.source]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu118"
-default = false
-secondary = false
-
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"