fixes #106

this is only a renaming of the previous commit

Author: EliHei2

src/segger/prediction/predict_parquet.py

@@ -6,8 +6,6 @@
 import torch.nn.functional as F
 import torch._dynamo
 import gc
-
-# import rmm
 import re
 import glob
 from pathlib import Path