Modification du train.py

.env

@@ -0,0 +1,4 @@
+AWS_ACCESS_KEY_ID=AKIA5TMZ4ZTASKBPWY36
+AWS_SECRET_ACCESS_KEY=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
+ARTIFACT_STORE_URI=s3://mlflow-server-dsfs38/
+BACKEND_STORE_URI=postgresql://neondb_owner:npg_F9QsrpxhyXz1@ep-small-math-ag82ufcb-pooler.c-2.eu-central-1.aws.neon.tech/neondb?sslmode=require&channel_binding=require

train.py

@@ -3,8 +3,11 @@ import pandas as pd
 from sklearn.datasets import load_iris
 from sklearn.linear_model import LogisticRegression
 from sklearn.model_selection import train_test_split
+from dotenv import load_dotenv
 import os
 
+load_dotenv()
+
 # Load Iris dataset
 iris = load_iris()