@@ -50,7 +50,7 @@ async def get(
5050 )
5151 row : RowProxy | None = await result .first ()
5252 if not row :
53- raise ComputationalRunNotFoundError ()
53+ raise ComputationalRunNotFoundError
5454 return CompRunsAtDB .from_orm (row )
5555
5656 async def list (
@@ -80,7 +80,7 @@ async def create(
8080 project_id : ProjectID ,
8181 cluster_id : ClusterID ,
8282 iteration : PositiveInt | None = None ,
83- metadata : RunMetadataDict | None ,
83+ metadata : RunMetadataDict ,
8484 use_on_demand_clusters : bool ,
8585 ) -> CompRunsAtDB :
8686 try :
@@ -102,13 +102,13 @@ async def create(
102102 .values (
103103 user_id = user_id ,
104104 project_uuid = f"{ project_id } " ,
105- cluster_id = cluster_id
106- if cluster_id != DEFAULT_CLUSTER_ID
107- else None ,
105+ cluster_id = (
106+ cluster_id if cluster_id != DEFAULT_CLUSTER_ID else None
107+ ) ,
108108 iteration = iteration ,
109109 result = RUNNING_STATE_TO_DB [RunningState .PUBLISHED ],
110- started = datetime .datetime .now (tz = datetime .timezone . utc ),
111- metadata = jsonable_encoder (metadata ) if metadata else None ,
110+ started = datetime .datetime .now (tz = datetime .UTC ),
111+ metadata = jsonable_encoder (metadata ),
112112 use_on_demand_clusters = use_on_demand_clusters ,
113113 )
114114 .returning (literal_column ("*" ))
@@ -146,7 +146,7 @@ async def set_run_result(
146146 ) -> CompRunsAtDB | None :
147147 values : dict [str , Any ] = {"result" : RUNNING_STATE_TO_DB [result_state ]}
148148 if final_state :
149- values .update ({"ended" : datetime .datetime .now (tz = datetime .timezone . utc )})
149+ values .update ({"ended" : datetime .datetime .now (tz = datetime .UTC )})
150150 return await self .update (
151151 user_id ,
152152 project_id ,
0 commit comments