@@ -18,7 +18,7 @@ def _require_mlflow() -> None:
         import mlflow  # noqa: F401
     except ImportError as e:
         raise click.ClickException(
-            "The 'mlflow' package is required. Install it with: pip install 'feast[mlflow]'"
+            "The 'mlflow' package is required. Install the 'feast[mlflow]' extra in your environment."
         ) from e
 
 
 
@@ -80,7 +80,7 @@ def _build_client() -> Any:
     except ImportError:
         raise ImportError(
             "mlflow package is not installed. "
-            "Install it with: pip install feast[mlflow]"
+            "Install the 'feast[mlflow]' extra in your environment."
         )
 
 
 
@@ -81,7 +81,7 @@ def sync_mlflow_dataset_to_feast(
     except ImportError as e:
         raise ImportError(
             "The 'mlflow' package is required for dataset sync. "
-            "Install it with: pip install 'feast[mlflow]'"
+            "Install the 'feast[mlflow]' extra in your environment."
         ) from e
 
     result = SyncResult()
Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ def _build_client() -> Any:`
`80`	`80`	`except ImportError:`
`81`	`81`	`raise ImportError(`
`82`	`82`	`"mlflow package is not installed. "`
`83`		`- "Install it with: pip install feast[mlflow]"`
	`83`	`+ "Install the 'feast[mlflow]' extra in your environment."`
`84`	`84`	`)`
`85`	`85`
`86`	`86`