CarbonIntensityProvider and ElectricityMaps implementation

ml-energy · Sep 30, 2024 · 1b10e7d · 1b10e7d
1 parent 50b2808
commit 1b10e7d
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 76 deletions.
diff --git a/tests/carbon/test_electricity_maps_carbon_intensity_provider.py b/tests/carbon/test_electricity_maps_carbon_intensity_provider.py
@@ -1,15 +1,15 @@
 from __future__ import annotations
 
-import requests
-import pytest
 import json
+import pytest
+import requests
 
 from unittest.mock import patch
 
 from zeus.carbon.carbon import (
     ElectrictyMapsClient,
     get_ip_lat_long,
-    CarbonIntensityNotFoundException,
+    CarbonIntensityNotFoundError,
 )
 
 
@@ -100,5 +100,5 @@ def test_get_current_carbon_intensity_no_response(mock_requests):
     assert latlong == (pytest.approx(42.2776), pytest.approx(-83.7409))
     provider = ElectrictyMapsClient(latlong)
 
-    with pytest.raises(CarbonIntensityNotFoundException):
+    with pytest.raises(CarbonIntensityNotFoundError):
         provider.get_current_carbon_intensity()
diff --git a/zeus/carbon/__init__.py b/zeus/carbon/__init__.py
diff --git a/zeus/carbon/carbon.py b/zeus/carbon/carbon.py
@@ -2,10 +2,10 @@
 
 from __future__ import annotations
 
-import requests
-import logging
 import abc
 import json
+import logging
+import requests
 
 logger = logging.getLogger(__name__)
 

diff --git a/zeus/carbon/carbon_intensity_provider.py b/zeus/carbon/carbon_intensity_provider.py
diff --git a/zeus/carbon/electricity_maps_carbon_intensity_provider.py b/zeus/carbon/electricity_maps_carbon_intensity_provider.py