ensure angles are always stored as a list

Author: sophiamaedler

src/scportrait/ml/transforms.py

@@ -18,7 +18,7 @@ def __init__(self, choices=4, include_zero=True):
             delta = (360 - angles[-1]) / 2
             angles = angles + delta
 
-        self.choices = angles
+        self.choices = angles.tolist()
 
     def __call__(self, tensor):
         angle = random.choice(self.choices)