Skip to content

Commit dc1c0df

Browse files
authored
Remove unnecessary comment
1 parent 87efc6c commit dc1c0df

File tree

1 file changed

+3
-10
lines changed
  • template/{{.input_root_dir}}/{{template `project_name_alphanumeric_underscore` .}}/validation/notebooks

1 file changed

+3
-10
lines changed

template/{{.input_root_dir}}/{{template `project_name_alphanumeric_underscore` .}}/validation/notebooks/ModelValidation.py.tmpl

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -312,26 +312,19 @@ data = rounded_taxi_data(data)
312312
# COMMAND ----------
313313

314314
{{ if (eq .input_include_feature_store `yes`) }}
315-
###########################################################################################
316-
# Temporary fix as FS model can't predict as pyfunc model #
317-
# MLflow evaluate can take a lambda function instead of model uri for model #
318-
# but not for baseline model it requires model_uri (baseline comparison set to false) #
319-
###########################################################################################
315+
# Temporary fix as FS model can't predict as a pyfunc model
316+
# MLflow evaluate can take a lambda function instead of a model uri for a model
317+
# but id does not work for the baseline model as it requires a model_uri (baseline comparison is set to false)
320318

321319
from databricks.feature_store import FeatureStoreClient
322320

323-
324321
def get_fs_model(df):
325322
fs_client = FeatureStoreClient()
326323
return (
327324
fs_client.score_batch(model_uri, spark.createDataFrame(df))
328325
.select("prediction")
329326
.toPandas()
330327
)
331-
332-
333-
###########################################################################################
334-
335328
{{ end }}
336329

337330
training_run = get_training_run(model_name, model_version)

0 commit comments

Comments
 (0)