Merge pull request #13 from alan-turing-institute/8-pytorch

Quick pull request resolving #8 now pytorch 1.13.1 is released

README.md

@@ -52,8 +52,6 @@ You can begin using the package code with an import command:
    poetry run pre-commit install --install-hooks
    ```
 
-3. Note that per issue [#8] you may encounter difficulties in installing pytorch with the current setup (i.e. if you are not running a mac with arm architecture). If this is the case, please update the URLs on lines 13 and 14 of pyproject.toml
-
 ### Testing
 
 1. To run tests:

pyproject.toml

@@ -10,11 +10,11 @@ packages = [{include = "modsim2", from="src"}]
 python = "^3.10"
 pre-commit = "^2.20.0"
 pytest = "^7.2.0"
-torch = {platform="macosx", url = "https://download.pytorch.org/whl/cpu/torch-1.13.0-cp310-none-macosx_11_0_arm64.whl"}
-torchvision = {platform="macosx", url="https://download.pytorch.org/whl/cpu/torchvision-0.14.0-cp310-cp310-macosx_11_0_arm64.whl"}
 pytorch-lightning = "^1.8.3.post1"
 lightning-bolts = "^0.6.0.post1"
 scikit-learn = "^1.1.3"
+torch = "^1.13.1"
+torchvision = "^0.14.1"
 
 [build-system]
 requires = ["poetry-core"]
@@ -24,6 +24,9 @@ build-backend = "poetry.core.masonry.api"
 pythonpath = [
     "src",
 ]
+filterwarnings = [
+    "ignore::pl_bolts.utils.stability.UnderReviewWarning",
+]
 
 [tool.black]
 line-length = 88