add opencv-python

climategan_wrapper.py

@@ -576,7 +576,11 @@ if __name__ == "__main__":
     cg = ClimateGAN("models/climategan")
 
     # check painter type
-    assert args.painter in {"climategan", "stable_diffusion", "both",}, (
+    assert args.painter in {
+        "climategan",
+        "stable_diffusion",
+        "both",
+    }, (
         f"Unknown painter {args.painter}. "
         + "Allowed values are 'climategan', 'stable_diffusion' and 'both'."
     )

requirements.txt

@@ -87,6 +87,7 @@ networkx
 numpy
 nvidia-ml-py3
 omegaconf
+opencv-python
 orjson
 packaging
 pandas