diff --git a/src/crewai/integrations/mlflow.py b/src/crewai/integrations/mlflow.py
index 352805749..4384c93f7 100644
--- a/src/crewai/integrations/mlflow.py
+++ b/src/crewai/integrations/mlflow.py
@@ -1,7 +1,6 @@
 """MLFlow integration utilities for CrewAI."""
 
 import logging
-from typing import Optional
 
 logger = logging.getLogger(__name__)
 
@@ -9,7 +8,7 @@ logger = logging.getLogger(__name__)
 def is_mlflow_available() -> bool:
     """Check if MLFlow is available."""
     try:
-        import mlflow
+        import mlflow  # noqa: F401
         return True
     except ImportError:
         return False
diff --git a/tests/integrations/test_mlflow.py b/tests/integrations/test_mlflow.py
index d8212a989..db9012195 100644
--- a/tests/integrations/test_mlflow.py
+++ b/tests/integrations/test_mlflow.py
@@ -1,7 +1,7 @@
 """Tests for MLFlow integration."""
 
 import pytest
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 import sys