corva-worker-python 2.2.1__tar.gz → 2.3.0__tar.gz

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corva-worker-python
-Version: 2.2.1
+Version: 2.3.0
 Summary: SDK for interacting with Corva
 Home-page: https://github.com/corva-ai/corva-worker-python
 Author: Jordan Ambra <jordan.ambra@corva.ai>, Mohammadreza Kamyab <m.kamyab@corva.ai>

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/corva_worker_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: corva-worker-python
-Version: 2.2.1
+Version: 2.3.0
 Summary: SDK for interacting with Corva
 Home-page: https://github.com/corva-ai/corva-worker-python
 Author: Jordan Ambra <jordan.ambra@corva.ai>, Mohammadreza Kamyab <m.kamyab@corva.ai>

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/setup.py

@@ -15,7 +15,7 @@ setup(
     name="corva-worker-python",
     author="Jordan Ambra <jordan.ambra@corva.ai>, Mohammadreza Kamyab <m.kamyab@corva.ai>",
     url="https://github.com/corva-ai/corva-worker-python",
-    version="2.2.1",
+    version="2.3.0",
     classifiers=classifiers,
     python_requires=">=3.13",
     description="SDK for interacting with Corva",

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/worker/partial_rerun_merge/merge.py

@@ -199,7 +199,7 @@ class PartialRerunMerge:
         if not dataset_progress:
             if not skip_progress:
                 return True
-            dataset_progress = DatasetProgress(0, -1, False, collection_name)
+            dataset_progress = DatasetProgress(0, -1, False, collection_name, "time")
 
         start_time = self.start_timestamp
         if dataset_progress.is_started():

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/worker/partial_rerun_merge/progress.py

@@ -22,6 +22,7 @@ class DatasetProgress:
         processed_timestamp (int): The timestamp when the dataset was processed.
         completed (bool): Whether the dataset processing is completed or not.
         dataset_name (str): The name of the dataset.
+        dataset_type (str): The type of the dataset. Not used here.
     """
 
     dataset_id: int

{corva_worker_python-2.2.1 → corva_worker_python-2.3.0}/worker/wellbore/factory.py

@@ -13,6 +13,10 @@ from worker.wellbore.wellbore import Wellbore
 MIN_LENGTH = 1  # ft
 
 
+def _remove_planning_drillstrings(drillstrings: List[dict]) -> List[dict]:
+    return [ds for ds in drillstrings if not ds.get("data", {}).get("planning", False)]
+
+
 def run_drillstring_and_create_wellbore(wits: dict, mud_flow_in: float = None) -> Union[Wellbore, None]:
     """
     Get the wits record and create a wellbore
@@ -64,6 +68,7 @@ def run_drillstring_and_create_wellbore(wits: dict, mud_flow_in: float = None) -
     drillstrings = api_worker.get(
         path="/v1/data/corva", collection="data.drillstring", asset_id=asset_id, query=query, sort=sort, limit=100
     ).data
+    drillstrings = _remove_planning_drillstrings(drillstrings)
     hole = add_open_holes(hole, drillstrings, smallest_cased_hole_diameter, hole_depth)
 
     mud_flow_in = get_mud_flow_in(mud_flow_in, wits)
@@ -160,6 +165,8 @@ def run_casingstring_and_create_wellbore(
     drill_sections = api_worker.get(
         path="/v1/data/corva", collection=drilling_data_source, asset_id=asset_id, sort=sort, limit=100
     ).data
+    if not is_design:
+        drill_sections = _remove_planning_drillstrings(drill_sections)
 
     # use data.well-sections to create hole
     if is_design: