From 9966281a0ba31cd0ac4f6100582c8723501617d9 Mon Sep 17 00:00:00 2001
From: Thomas Vuillaume <thomas.vuillaume@gmail.com>
Date: Wed, 22 Jun 2022 10:23:24 +0200
Subject: [PATCH] Testing env with pip install torch

Might solve #35 but can't be tested in CI
---
 environment.yml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/environment.yml b/environment.yml
index 4845ed0..7fdbead 100644
--- a/environment.yml
+++ b/environment.yml
@@ -26,9 +26,6 @@ dependencies:
   - line_profiler
   - memory_profiler
   - snakeviz
-  - pytorch
-  - torchvision
-  - torchaudio
   - plotly
   - nbmake
   - h5py
@@ -39,3 +36,6 @@ dependencies:
     - build
     - tensorboard
     - pytorch-lightning
+    - torch
+    - torchvision
+    - torchaudio