more formatting fixes

Author: C00ldudeNoonan

macro_agents/src/macro_agents/defs/agents/dspy_evaluation.py

@@ -1,7 +1,6 @@
 import dspy
 from typing import Optional, Dict, Any, List
 import json
-import re
 from datetime import datetime, timedelta
 import dagster as dg
 from pydantic import Field

macro_agents/src/macro_agents/defs/agents/model_improvement_pipeline.py

@@ -108,23 +108,6 @@ def create_training_dataset(
             params=[start_date.strftime("%Y-%m-%d"), end_date.strftime("%Y-%m-%d")],
         )
 
-        # Get market trend analysis
-        trend_query = """
-        SELECT 
-            analysis_content,
-            analysis_timestamp
-        FROM enhanced_economic_cycle_analysis
-        WHERE analysis_type = 'enhanced_market_trends'
-        AND analysis_timestamp BETWEEN ? AND ?
-        ORDER BY analysis_timestamp
-        """
-
-        trend_df = md_resource.execute_query(
-            trend_query,
-            read_only=True,
-            params=[start_date.strftime("%Y-%m-%d"), end_date.strftime("%Y-%m-%d")],
-        )
-
         # Get backtest results for actual returns
         backtest_query = """
         SELECT 

macro_agents/src/macro_agents/defs/resources/motherduck.py

@@ -5,7 +5,6 @@
 from pydantic import Field
 import os
 import io
-import re
 
 
 class MotherDuckResource(dg.ConfigurableResource):
@@ -61,7 +60,7 @@ def get_connection(self) -> duckdb.DuckDBPyConnection:
                     conn = duckdb.connect(self.db_connection)
                     conn.commit()
                     return conn
-                except:
+                except Exception:
                     raise e
             raise
 

macro_agents/tests/test_integration.py

@@ -87,10 +87,6 @@ def test_analyzer_integration(self, mock_configure, mock_lm):
                 md_token="test_token", environment="dev", local_path=tmp_file.name
             )
 
-            analyzer = EconomicAnalyzer(
-                model_name="gpt-4-turbo-preview", openai_api_key="test_key"
-            )
-
             # Create test data
             test_df = pl.DataFrame(
                 {

macro_agents/tests/test_schedules.py

@@ -121,7 +121,6 @@ def test_sensor_monday_trigger(self, mock_datetime):
         )
 
         sensor = create_data_freshness_sensor()
-        context = Mock()
 
         # This would need to be tested with actual sensor execution
         # For now, just verify the sensor is created
@@ -138,7 +137,6 @@ def test_sensor_monthly_trigger(self, mock_datetime):
         )
 
         sensor = create_data_freshness_sensor()
-        context = Mock()
 
         # This would need to be tested with actual sensor execution
         # For now, just verify the sensor is created