Update loader

src/data/waxal_loader.py

@@ -64,6 +64,7 @@ class WaxalDataLoader:
             split=split,
             token=self.hf_token,
             streaming=streaming,
+            trust_remote_code=True
         )
         if streaming:
             ds = ds.shuffle(seed=42, buffer_size=1000)