Use experiment metadata to see experiment id in metrics_analysis.py

This commit is contained in:
afik.cohen 2024-05-07 16:59:34 -07:00
parent ee98254d2f
commit 22c7466e53

View file

@ -13,7 +13,7 @@ def load_data(experiment_id: str, database_uri: str = os.environ.get("DATABASE_U
query = """
SELECT * FROM optimization_features
WHERE (trace_id LIKE %s OR trace_id LIKE %s)
AND experiment_id = %s
AND experiment_metadata->>'id' = %s
"""
return pd.read_sql_query(
query,