test env var setup GHA

Author: divakaivan

.github/workflows/train_model.yaml

@@ -6,7 +6,7 @@ on:
       - train_model
 
 jobs:
-  build_and_test:
+  train_model:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
@@ -39,7 +39,7 @@ jobs:
 
       - name: Train Model
         env:
-          MLFLOW_TRACKING_URI_DEV: ${{ secrets.MLFLOW_TRACKING_URI_DEV }}
+          MLFLOW_TRACKING_URI: ${{ secrets.MLFLOW_TRACKING_URI }}
           GOOGLE_APPLICATION_CREDENTIALS: ${{ github.workspace }}/gcp-key.json
         run: |
           uv run python scripts/3_train_model.py \

scripts/3_train_model.py

@@ -3,19 +3,15 @@
 import os
 
 import mlflow
-from dotenv import load_dotenv
 
 from make_data.gcs_connector import GCSConnector
 from make_model.model_trainer import ModelTrainer
 from project_config import ProjectConfig, Tags
 
-load_dotenv()
-
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-mlflow.set_tracking_uri(os.environ["MLFLOW_TRACKING_URI"])
-logging.info(os.environ("MLFLOW_TRACKING_URI"))
+mlflow.set_tracking_uri(os.getenv("MLFLOW_TRACKING_URI"))
 
 parser = argparse.ArgumentParser()
 parser.add_argument(