Skip to content

Commit a39effc

Browse files
committed
more group fixes
1 parent 1b3a1b4 commit a39effc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

code/utils/group.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def plot_person(self):
7676

7777
self._plot_stacked_bar(df_all,
7878
title="Normalized Average Activity Composition by Subject (All Sessions)",
79-
filename="avg_plot_all")
79+
filename="avg_plot_all.html")
8080

8181
def plot_session(self):
8282
durations = []
@@ -122,7 +122,7 @@ def plot_session(self):
122122
group_df,
123123
title=f"Average Activity Composition — Session {session.upper()}",
124124
y_key="Subject",
125-
filename=f"avg_plot_{session}"
125+
filename=f"avg_plot_{session}.html"
126126
)
127127

128128
def _plot_stacked_bar(self, df, title, filename, y_key="Subject"):
@@ -160,5 +160,5 @@ def _plot_stacked_bar(self, df, title, filename, y_key="Subject"):
160160
showlegend=True,
161161
autosize=True
162162
)
163-
163+
os.makedirs(self.path, exist_ok=True)
164164
fig.write_html(os.path.join(self.path, filename))

0 commit comments

Comments
 (0)