refactored prod_shares

Grid2Kpi.iml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

grid2kpi/episode/EpisodeAnalytics.py

@@ -272,6 +272,15 @@ def _env_actions_as_df(self):
 
         return hazards, maintenances
 
+    def get_prod_types(self):
+        types = self.observation_space.gen_type
+        ret = {}
+        if types is None:
+            return ret
+        for (idx, name) in enumerate(self.prod_names):
+            ret[name] = types[idx]
+        return ret
+
 
 class Test():
     def __init__(self):

grid2kpi/episode/EpisodeTrace.py

@@ -24,7 +24,8 @@ def get_total_overflow_ts(episode):
     return df
 
 
-def get_prod_share_trace(episode, prod_types):
+def get_prod_share_trace(episode):
+    prod_types = episode.get_prod_types()
     prod_type_values = list(prod_types.values()) if len(
         prod_types.values()) > 0 else []
 
@@ -106,7 +107,7 @@ def get_all_prod_trace(episode, prod_types, selection):
         if name in selection:
             trace.append(go.Scatter(
                 x=prod_with_type[prod_with_type.prod_type.values ==
-                                 name]['timestamp'].unique(),
+                                 name]['timestamp'].drop_duplicates(),
                 y=prod_with_type[prod_with_type.prod_type.values == name].groupby(['timestamp'])[
                     'value'].sum(),
                 name=name