From 2c00f17bf6c75cbc7dba66bb849aec873b36e6a9 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 12 Aug 2025 16:28:15 +0000
Subject: [PATCH] Fix lint errors: remove unused imports
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Remove unused typing.Optional import
- Remove unused unittest.mock.MagicMock import
- Add noqa comment for necessary mlflow import in availability check

Co-Authored-By: João <joao@crewai.com>
---
 src/crewai/integrations/mlflow.py | 3 +--
 tests/integrations/test_mlflow.py | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

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