From 0069b1efa85c9287a60ed4b0135709835809149b Mon Sep 17 00:00:00 2001
From: Yolan <yolanother@gmail.com>
Date: Sun, 22 Dec 2024 17:50:01 -0800
Subject: [PATCH] Finish removing openexr dep

---
 __init__.py | 19 ++-----------------
 1 file changed, 2 insertions(+), 17 deletions(-)

diff --git a/__init__.py b/__init__.py
index e2474cd..9c827ad 100644
--- a/__init__.py
+++ b/__init__.py
@@ -3,29 +3,14 @@ import sys
 import folder_paths
 import numpy as np
 import re
-import tempfile
+import cv2
 
 sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
 
 
 class SaveImageARGB16PNG:
     def __init__(self):
-        try:
-            import OpenEXR
-            import Imath
-            self.OpenEXR = OpenEXR
-            self.Imath = Imath
-            self.use_openexr = True
-            print("AARON: Using OpenEXR!")
-        except ImportError:
-            print("No OpenEXR module found, trying OpenCV...")
-            self.use_openexr = False
-            try:
-                os.environ["OPENCV_IO_ENABLE_OPENEXR"] = "1"
-                import cv2
-                self.cv2 = cv2
-            except ImportError:
-                raise ImportError("No OpenEXR or OpenCV module found, can't save EXR")
+        self.cv2 = cv2
 
         self.output_dir = folder_paths.get_output_directory()
         self.type = "output"