linting

Author: j-so

diabetes_regression/scoring/parallel_batchscore.py

@@ -60,9 +60,9 @@ def parse_args() -> List[str]:
     model_name = model_name_param[0][1]
 
     model_version_param = [
-    (sys.argv[idx], sys.argv[idx + 1])
-    for idx, itm in enumerate(sys.argv)
-    if itm == "--model_version"
+        (sys.argv[idx], sys.argv[idx + 1])
+        for idx, itm in enumerate(sys.argv)
+        if itm == "--model_version"
     ]
 
     if len(model_version_param) == 0:
@@ -109,7 +109,7 @@ def init():
 
         model_filter = parse_args()
         amlmodel = get_model(
-            model_name=env.model_filter[0],
+            model_name=model_filter[0],
             model_version=model_filter[1],
             tag_name=model_filter[2],
             tag_value=model_filter[3])

diabetes_regression/util/model_helper.py

@@ -24,7 +24,7 @@ def get_current_workspace() -> Workspace:
 
 def get_model(
     model_name: str,
-    model_version: int = None, # If none, return latest model
+    model_version: int = None,  # If none, return latest model
     tag_name: str = None,
     tag_value: str = None,
     aml_workspace: Workspace = None
@@ -36,18 +36,22 @@ def get_model(
     Parameters:
     aml_workspace (Workspace): aml.core Workspace that the model lives.
     model_name (str): name of the model we are looking for
-    (optional) model_version (str): version of the model. Returns latest if not provided.
+    (optional) model_version (str): model version. Latest if not provided.
     (optional) tag (str): the tag value & name the model was registered under.
 
     Return:
     A single aml model from the workspace that matches the name and tag.
     """
     if aml_workspace is None:
-            print("No workspace defined - using current experiment workspace.")
-            aml_workspace = get_current_workspace()
+        print("No workspace defined - using current experiment workspace.")
+        aml_workspace = get_current_workspace()
 
     if tagname is not None and tagvalue is not None:
-        model = Model(aml_workspace, name=model_name, version=model_version, tags=[[tag_name, tag_value]])
+        model = AMLModel(
+            aml_workspace,
+            name=model_name,
+            version=model_version,
+            tags=[[tag_name, tag_value]])
     elif (tagname is None and tagvalue is not None) or (
         tagvalue is None and tagname is not None
     ):
@@ -56,5 +60,5 @@ def get_model(
             + "or excluded"  # NOQA: E501
         )
     else:
-        model = Model(aml_workspace, name=env.model_name, version=env.model_version)
+        model = AMLModel(aml_workspace, name=env.model_name, version=env.model_version) # NOQA: E501
     return model

ml_service/pipelines/diabetes_regression_build_parallel_batchscore_pipeline.py

@@ -61,6 +61,7 @@ def parse_args() -> Namespace:
     args = parser.parse_args()
     return args
 
+
 def get_or_create_datastore(
     datastorename: str, ws: Workspace, env: Env, input: bool = True
 ) -> Datastore:

ml_service/pipelines/diabetes_regression_verify_train_pipeline.py

@@ -54,7 +54,11 @@ def main():
     try:
         tag_name = 'BuildId'
         model = get_model(
-            model_name=model_name, tag_name=tag_name, tag_value=build_id, aml_workspace=exp.workspace)
+            model_name=model_name,
+            tag_name=tag_name,
+            tag_value=build_id,
+            aml_workspace=exp.workspace)
+
         if (model is not None):
             print("Model was registered for this build.")
         if (model is None):