Review comments

gnosis · Feb 14, 2024 · d1b8775 · d1b8775
1 parent 8d31e96
commit d1b8775
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/monitor/monitor.py b/examples/monitor/monitor.py
@@ -1,5 +1,6 @@
 from datetime import datetime, timedelta
 from zoneinfo import ZoneInfo
+import streamlit as st
 
 from prediction_market_agent_tooling.markets.manifold import get_authenticated_user
 from prediction_market_agent_tooling.monitor.markets.manifold import (
@@ -14,4 +15,5 @@
         start_time=start_time.astimezone(ZoneInfo("UTC")),
         manifold_user_id=get_authenticated_user().id,
     )
+    st.set_page_config(layout="wide")  # Best viewed with a wide screen
     monitor_agent(agent)
diff --git a/prediction_market_agent_tooling/monitor/monitor.py b/prediction_market_agent_tooling/monitor/monitor.py
@@ -4,14 +4,13 @@
 import pandas as pd
 import streamlit as st
 import typing as t
-from zoneinfo import ZoneInfo
 
 from prediction_market_agent_tooling.markets.data_models import ResolvedBet
 
 
 class DeployedAgent(BaseModel):
     name: str
-    start_time: datetime = datetime.now().astimezone(tz=ZoneInfo("UTC"))
+    start_time: datetime = datetime.utcnow()
     end_time: t.Optional[datetime] = None
 
     def get_resolved_bets(self) -> list[ResolvedBet]:
@@ -31,7 +30,6 @@ def monitor_agent(agent: DeployedAgent) -> None:
     }
     bets_df = pd.DataFrame(bets_info).sort_values(by="Resolved Time")
 
-    st.set_page_config(layout="wide")
     st.title(f"Monitoring Agent: '{agent.name}'")
 
     # Metrics