minor fix

powerapi-ng · May 5, 2020 · 801dcb3 · 801dcb3
1 parent e2cc223
commit 801dcb3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyJoules/__init__.py b/pyJoules/__init__.py
@@ -21,4 +21,4 @@
 from .energy_sample import EnergySample
 
 
-__version__ = '0.2.4'
+__version__ = '0.2.4.1'
diff --git a/pyJoules/energy_device/energy_device_factory.py b/pyJoules/energy_device/energy_device_factory.py
@@ -26,6 +26,7 @@
 try :
     from .nvidia_device import NvidiaGPUDevice
 except ImportError: 
+    NvidiaGPUDevice=None
     logging.warning(f'pynvml not found you can\'t use NVIDIA devices') 
 
 from ..exception import NoSuchEnergyDeviceError
@@ -37,7 +38,9 @@ class EnergyDeviceFactory:
 
     @staticmethod
     def _gen_all_available_domains() -> List[EnergyDevice]:
-        available_api = [RaplDevice, NvidiaGPUDevice]
+        available_api = [RaplDevice]
+        if NvidiaGPUDevice is not None : 
+            available_api.append(NvidiaGPUDevice)
         available_domains = []
         for api in available_api:
             try:
Original file line number	Diff line number	Diff line change
Expand Up		@@ -21,4 +21,4 @@
		from .energy_sample import EnergySample


		__version__ = '0.2.4'
		__version__ = '0.2.4.1'