Skip to content

Commit 6df1be9

Browse files
committed
Add a print
1 parent f296258 commit 6df1be9

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

pufferlib/pufferl.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1415,6 +1415,9 @@ def eval_womd(env_name, args=None, vecenv=None, policy=None):
14151415
cols = ["map_name"] + [col for col in df.columns if col != "map_name"]
14161416
df = df[cols]
14171417

1418+
num_unique_maps = len(df["map_name"].unique())
1419+
print(f"The logs contain {num_unique_maps} unique maps.")
1420+
14181421
# For now I do that, but ofc we should think at a way to manage files properly
14191422
df.to_csv("results.csv")
14201423

0 commit comments

Comments
 (0)