Update markov_wrapper.py

Fixed missing data bug
prio-data · Dec 14, 2023 · bf2753f · bf2753f
1 parent 023a710
commit bf2753f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Tools/new_markov/markov_wrapper.py b/Tools/new_markov/markov_wrapper.py
@@ -12,6 +12,7 @@
 qs2 = Queryset(qs,'country_month')
 
 data = qs2.fetch()
+data = data.fillna(0)
 data.to_parquet('tmps/tmp_data.parquet')
 
 pd.DataFrame({'s':steps}).to_parquet('tmps/tmp_steps.parquet')