Fixed groupby error

experiment_impact_tracker/gpu/nvidia.py

@@ -152,6 +152,9 @@ def get_nvidia_gpu_power(pid_list, logger=None, **kwargs):
     out_str_final = re.sub("\s+\n", "\n", out_str_final)  # else pd will mis-align
     out_str_final = out_str_final.strip()
     df = pd.read_csv(StringIO(out_str_final), engine="python", delimiter="\t")
+    try:
+        process_percentage_used_gpu = df.groupby(["gpu", "pid"])[['sm', 'mem', 'enc', 'dec']].mean().reset_index()
+    except:  # Older pandas versions
         process_percentage_used_gpu = df.groupby(["gpu", "pid"]).mean().reset_index()
 
     p = Popen(["nvidia-smi", "-q", "-x"], stdout=PIPE)