Skip to content

Commit 8114684

Browse files
fix: Fixed client use to match diracX PR
1 parent 535eca7 commit 8114684

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/DIRAC/WorkloadManagementSystem/FutureClient/PilotManagerClient.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,18 @@ class PilotManagerClient(FutureClient):
66
@convertToReturnValue
77
def addPilotReferences(self, pilot_references, VO, gridType="DIRAC", pilot_stamps_dict={}):
88
with DiracXClient() as api:
9-
pilot_stamps = pilot_stamps_dict.values()
9+
pilot_stamps = list(pilot_stamps_dict.values())
1010
pilot_ref_dict = dict(zip(pilot_stamps, pilot_references))
1111

1212
# We will move toward a stamp as identifier for the pilot
1313
return api.pilots.add_pilot_stamps(
14-
{"pilot_stamps": pilot_stamps, "vo": VO, "grid_type": gridType, "pilot_references": pilot_ref_dict} # type: ignore
14+
pilot_stamps=pilot_stamps, vo=VO, grid_type=gridType, pilot_references=pilot_ref_dict
1515
)
1616

1717
def set_pilot_field(self, pilot_stamp, values_dict):
1818
with DiracXClient() as api:
1919
values_dict["PilotStamp"] = pilot_stamp
20-
return api.pilots.update_pilot_fields({"pilot_stamps_to_fields_mapping": [values_dict]}) # type: ignore
20+
return api.pilots.update_pilot_fields(pilot_stamps_to_fields_mapping=[values_dict]) # type: ignore
2121

2222
@convertToReturnValue
2323
def setPilotStatus(self, pilot_stamp, status, destination=None, reason=None, grid_site=None, queue=None):

0 commit comments

Comments
 (0)