Get rid of Python class_keyword attribute and fix parameter sending for WBS

This commit is contained in:
Gaute Lindkvist
2020-03-03 14:58:36 +01:00
parent 554f9a1758
commit 76019e7776
5 changed files with 50 additions and 34 deletions

View File

@@ -826,7 +826,7 @@ def export_property(
))
@add_method(Case)
def create_well_bore_stability_plot(self, well_path, time_step, wbs_parameters=None):
def create_well_bore_stability_plot(self, well_path, time_step, parameters=None):
""" Create a new well bore stability plot
Arguments:
@@ -837,9 +837,9 @@ def create_well_bore_stability_plot(self, well_path, time_step, wbs_parameters=N
A new plot object
"""
pb2_parameters = None
if wbs_parameters is not None:
assert(isinstance(wbs_parameters, WbsParameters))
pb2_parameters = wbs_parameters.pb2_object()
if parameters is not None:
assert(isinstance(parameters, WbsParameters))
pb2_parameters = parameters.pb2_object()
plot_result = self._execute_command(createWellBoreStabilityPlot=Cmd.CreateWbsPlotRequest(caseId=self.id,
wellPath=well_path,