atlasai-dstoolkit-client 0.0.16__py3-none-any.whl → 0.0.18__py3-none-any.whl

atlasai/toolkit/workflows.py

@@ -47,7 +47,7 @@ class Workflow:
     def __init__(self, id=None):
         # Allow the selection of an existing workflow and check the logs.
         self.id = id
-        self.results = []
+        self._run_results = []
 
     @property
     def status(self):
@@ -157,6 +157,7 @@ class Electrification(Workflow):
         self.aoi_path = None
         self.aoi = None
         self.output_path = None
+        self.start_date = None
 
         super().__init__(id)
 
@@ -166,11 +167,12 @@ class Electrification(Workflow):
     def __str__(self):
         return f'Electrification({self.aoi_name})'
 
-    def configure(self, aoi_name=None, aoi_path=None, aoi=None, output_path=None):
+    def configure(self, aoi_name=None, aoi_path=None, aoi=None, start_date=None, output_path=None):
         self.aoi_name = aoi_name or self.aoi_name
         self.aoi_path = aoi_path or self.aoi_path
         self.aoi = aoi or self.aoi
         self.output_path = output_path or self.output_path
+        self.start_date = start_date or self.start_date
 
     def _config(self):
         cfg = {}
@@ -178,7 +180,8 @@ class Electrification(Workflow):
                 ('aoi_name', self.aoi_name),
                 ('aoi_geojson', self.aoi),
                 ('aoi_geojson_uri', self.aoi_path),
-                ('output_bucket', self.output_path)
+                ('output_bucket', self.output_path),
+                ('start_date', self.start_date)
         ):
             if value:
                 cfg[name] = value
@@ -191,9 +194,9 @@ class Electrification(Workflow):
         if not self.aoi_path and not self.aoi:
             raise Exception('`aoi_path` or `aoi` must be specified.')
 
-    def _results(self, save_to=None):
-        if self.results:
-            return self.results
+    def _results(self, save_to=None, force=False):
+        if self._run_results and not force:
+            return self._run_results
 
         results = []
         if not self.id:
@@ -211,7 +214,7 @@ class Electrification(Workflow):
             for fut in as_completed(fut_map):
                 results.append(fut.result())
                 pbar.update(1)
-        self.results = results
+        self._run_results = results
         return results
 
     def _download_file(self, name, url, to=None):
@@ -233,8 +236,9 @@ class Electrification(Workflow):
         return local_path
 
     def preview(self, from_=None, to_=None, selection=None, limit=None):
-        if not self.results:
+        if not self._run_results:
             raise Exception('No result to be displayed. Run the workflow first and wait for its completion.')
+
         _results = []
         if selection is None:
             selection = []
@@ -247,7 +251,7 @@ class Electrification(Workflow):
         if to_:
             to_ = datetime.strptime(to_, "%Y-%m-%d")
 
-        for f in sorted(self.results):
+        for f in sorted(self._run_results):
             yyyy, mm = f.split('_')[-1].replace('.tif', '').split('-')
             dt = datetime(year=int(yyyy), month=int(mm), day=1)
             if selection:

{atlasai_dstoolkit_client-0.0.16.dist-info → atlasai_dstoolkit_client-0.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: atlasai-dstoolkit-client
-Version: 0.0.16
+Version: 0.0.18
 Summary: UNKNOWN
 Home-page: UNKNOWN
 Author: AtlasAI SWE

{atlasai_dstoolkit_client-0.0.16.dist-info → atlasai_dstoolkit_client-0.0.18.dist-info}/RECORD

@@ -10,9 +10,9 @@ atlasai/toolkit/model.py,sha256=RUe0HbDpzvHOV9A4rzG3PgN9boMWDHQ2tR7IKHXzbx8,4126
 atlasai/toolkit/output.py,sha256=FyDjrpVlbrEyfHfwOpxp8H57jx_qXahDjO1qpHIeuYM,473
 atlasai/toolkit/requests.py,sha256=X86nIo07hAjUlilZcZ1lV8RB7KOsTKbTGtcY_SpFEXY,1223
 atlasai/toolkit/utils.py,sha256=lYh3P2XOshRgHCjFeXJ0FOJWQW64sddgx8c2kL6Wqwc,1566
-atlasai/toolkit/workflows.py,sha256=u17Hl-tIgyoB_-dA23g5h2o900fcsyj3H1uVVbZ4Mao,9392
-atlasai_dstoolkit_client-0.0.16.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-atlasai_dstoolkit_client-0.0.16.dist-info/METADATA,sha256=cC7U97hGZ68zR7Gapl1ZXPR2R2Q2Y4Xyp7--UtfdOnM,1547
-atlasai_dstoolkit_client-0.0.16.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-atlasai_dstoolkit_client-0.0.16.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
-atlasai_dstoolkit_client-0.0.16.dist-info/RECORD,,
+atlasai/toolkit/workflows.py,sha256=UXkNP84P-EExLtTiRGiO2PN0ZckCdf871GUhbIJtTvw,9603
+atlasai_dstoolkit_client-0.0.18.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+atlasai_dstoolkit_client-0.0.18.dist-info/METADATA,sha256=I91iQj6v4fkVVKWzvgw0XjiYTKrc45WiPw6KFPyK9po,1547
+atlasai_dstoolkit_client-0.0.18.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+atlasai_dstoolkit_client-0.0.18.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
+atlasai_dstoolkit_client-0.0.18.dist-info/RECORD,,