Cast block name to str

3 years ago · c2d1e52fe6
parent c2e3a1a484
commit c2d1e52fe6
1 changed files with 2 additions and 2 deletions
--- a/probabilistic-analysis/probabilistic_assessment.py
+++ b/probabilistic-analysis/probabilistic_assessment.py
@ -486,8 +486,8 @@ def process(beach_name, beach_scenario, n_runs, start_year, end_year,
                df_out = df_out.assign(segment_gaps=False)
                for b, p, in zip(segment_gaps['block'],
                                 segment_gaps['profile']):
-                    idx = (df_out['block'] == str(b)) & (df_out['profile']
-                                                         == p)
+                    idx = (str(df_out['block']) == str(b)) & (df_out['profile']
+                                                              == p)
                    df_out.loc[idx, 'segment_gaps'] = True

                # Specify which profiles to plot