feat: create an abstract grid class for model enclosed in CSV files (#…

…627)
Breakthrough-Energy · Jun 3, 2022 · 3a014f5 · 3a014f5
1 parent 2577aac
commit 3a014f5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/powersimdata/input/abstract_grid.py b/powersimdata/input/abstract_grid.py
@@ -31,6 +31,10 @@ def __init__(self):
         self.branch = pd.DataFrame()
         self.storage = storage_template()
 
+
+class AbstractGridCSV(AbstractGrid):
+    """Grid Builder."""
+
     def _set_data_loc(self, top_dirname):
         """Sets data location.
 

diff --git a/powersimdata/network/hifld/model.py b/powersimdata/network/hifld/model.py
@@ -1,10 +1,10 @@
 import os
 
-from powersimdata.input.abstract_grid import AbstractGrid
+from powersimdata.input.abstract_grid import AbstractGridCSV
 from powersimdata.network.constants.storage import get_storage
 
 
-class HIFLD(AbstractGrid):
+class HIFLD(AbstractGridCSV):
     """HIFLD network.
 
     :param str/iterable interconnect: interconnect name(s).

diff --git a/powersimdata/network/usa_tamu/model.py b/powersimdata/network/usa_tamu/model.py
@@ -1,10 +1,10 @@
 import os
 
-from powersimdata.input.abstract_grid import AbstractGrid
+from powersimdata.input.abstract_grid import AbstractGridCSV
 from powersimdata.network.constants.storage import get_storage
 
 
-class TAMU(AbstractGrid):
+class TAMU(AbstractGridCSV):
     """TAMU network.
 
     :param str/iterable interconnect: interconnect name(s).