Fixed env.yaml

hpvm/env.yaml

@@ -4,6 +4,7 @@ channels:
   - conda-forge
 dependencies:
   - Cython
+  - imgaug
   - jinja2=2.11
   - matplotlib=3.3
   - networkx=2.5
@@ -28,3 +29,5 @@ dependencies:
     - onnx-simplifier==0.3
     - torchnet==0.0.4
     - pexpect
+    - mean_average_precision
+    - pytorch-lightning==1.3
\ No newline at end of file