fixing test

AgnostiqHQ · Oct 13, 2023 · b487f58 · b487f58
1 parent 3262dfb
commit b487f58
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/covalent/_results_manager/result.py b/covalent/_results_manager/result.py
@@ -442,7 +442,7 @@ def _update_node(
         sublattice_result: "Result" = None,
         stdout: str = None,
         stderr: str = None,
-        qelectron_data_exists: bool = False,
+        qelectron_data_exists: bool = None,
     ) -> None:
         """
         Update the node result in the transport graph.
@@ -501,7 +501,7 @@ def _update_node(
         if stderr is not None:
             self.lattice.transport_graph.set_node_value(node_id, "stderr", stderr)
 
-        if qelectron_data_exists:
+        if qelectron_data_exists is not None:
             self.lattice.transport_graph.set_node_value(
                 node_id, "qelectron_data_exists", qelectron_data_exists
             )

diff --git a/covalent/_shared_files/schemas/electron.py b/covalent/_shared_files/schemas/electron.py
@@ -17,7 +17,7 @@
 """FastAPI models for /api/v1/resultv2 endpoints"""
 
 from datetime import datetime
-from typing import Dict, Optional, Union
+from typing import Dict, Optional
 
 from pydantic import BaseModel, validator
 
@@ -97,7 +97,7 @@ class ElectronMetadata(BaseModel):
     name: str
     executor: str
     executor_data: dict
-    qelectron_data_exists: Union[None, bool]
+    qelectron_data_exists: bool
     sub_dispatch_id: Optional[str] = None
     status: StatusEnum
     start_time: Optional[datetime] = None

diff --git a/covalent/_workflow/transport.py b/covalent/_workflow/transport.py
@@ -83,7 +83,7 @@ def encode_metadata(metadata: dict) -> dict:
             encoded_metadata["triggers"] = metadata["triggers"]
 
     # qelectron_data_exists
-    encoded_metadata["qelectron_data_exists"] = None
+    encoded_metadata["qelectron_data_exists"] = False
 
     return encoded_metadata
 

diff --git a/covalent_dispatcher/_db/update.py b/covalent_dispatcher/_db/update.py
@@ -74,7 +74,7 @@ def _node(
     stderr: str = None,
     sub_dispatch_id=None,
     sublattice_result=None,
-    qelectron_data_exists: bool = False,
+    qelectron_data_exists: bool = None,
 ) -> bool:
     """
     Update the node result in the transport graph.

diff --git a/covalent_ui/api/v1/database/schema/electron.py b/covalent_ui/api/v1/database/schema/electron.py
@@ -127,7 +127,7 @@ class Electron(Base):
     cancel_requested = Column(Boolean, nullable=False, default=False)
 
     # Flag that indicates if qelectron data exists in the electron
-    qelectron_data_exists = Column(Boolean, nullable=False)
+    qelectron_data_exists = Column(Boolean, nullable=False, default=False)
 
     # Timestamps
     created_at = Column(DateTime, nullable=False, server_default=func.now())