Fix FutureWarning in CategoricalTensorMapper.

pyg-team · May 20, 2024 · 563f93a · 563f93a
1 parent ef10a69
commit 563f93a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_frame/data/mapper.py b/torch_frame/data/mapper.py
@@ -110,7 +110,7 @@ def forward(
 
     def backward(self, tensor: Tensor) -> pd.Series:
         index = tensor.cpu().numpy()
-        ser = pd.Series(self.categories[index].index)
+        ser = pd.Series(self.categories.iloc[index].index)
         ser[index < 0] = None
         return ser