Update tutorial and fix plotting bug

1 year ago · eca4cae298
parent 47a67f6665
commit eca4cae298
2 changed files with 127 additions and 74 deletions
--- a/docs/tutorial/soil_tutorial.ipynb
+++ b/docs/tutorial/soil_tutorial.ipynb
--- a/soil/analysis.py
+++ b/soil/analysis.py
@ -21,8 +21,10 @@ def plot(env, agent_df=None, model_df=None, steps=False, ignore=["agent_count",
        except UserWarning:
            print("No agent dataframe provided and no agent reporters found. Skipping agent plot.", file=sys.stderr)
            return
-    agent_df.unstack().apply(lambda x: x.value_counts(),
-                                axis=1).fillna(0).plot(ax=ax, secondary_y=True);
+    if not agent_df.empty:
+        agent_df.unstack().apply(lambda x: x.value_counts(),
+                                 axis=1).fillna(0).plot(ax=ax,
+                                                        secondary_y=True)


 Results = namedtuple("Results", ["config", "parameters", "env", "agents"])