mlflow / mlflow-export-import

Apache License 2.0
110 stars 70 forks source link

unexpected keywork while running databricks_notebooks/bulk/Export_All.py in databricks #170

Open prasiit92 opened 4 months ago

prasiit92 commented 4 months ago

Below is the error that I got while running the notebook: output_dir: /Shared/MLFlowExport stages: Production,None export_latest_versions: False run_start_date: None export_permissions: False export_deleted_runs: False export_version_model: False notebook_formats: ['SOURCE'] use_threads: False TypeError: export_all() got an unexpected keyword argument 'stages'

TypeError Traceback (most recent call last) File :70 66 # COMMAND ---------- 68 from mlflow_export_import.bulk.export_all import export_all ---> 70 export_all( 71 output_dir = output_dir, 72 stages = stages, 73 export_latest_versions = export_latest_versions, 74 run_start_time = run_start_date, 75 export_permissions = export_permissions, 76 export_deleted_runs = export_deleted_runs, 77 export_version_model = export_version_model, 78 notebook_formats = notebook_formats, 79 use_threads = use_threads 80 )

I am facing the same issue while running export_expirements as well

andremesarovic commented 2 months ago

Need more details such as what command you are running and stack trace. Make sure you install from github and not from PyPI.