change dep

scaleoutsystems · Sep 2, 2024 · 2fe040a · 2fe040a
1 parent 70e6705
commit 2fe040a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mnist-pytorch-DPSGD/client/python_env.yaml b/examples/mnist-pytorch-DPSGD/client/python_env.yaml
@@ -4,7 +4,7 @@ build_dependencies:
   - setuptools
   - wheel
 dependencies:
-  - torch==2.3.1
-  - torchvision==0.18.1
+  - torch
+  - torchvision
   - fedn
   - opacus