Fix data type (#26)

Fixes # da ta type error

### Description

File "/localhome/local-mingxueg/VISTA/scripts/utils/trans_utils.py",
line 352, in __call__
pred[pred == frac] = data["label_prompt"][i - 1].to(pred.dtype)
AttributeError: 'int' object has no attribute 'to'

### Types of changes

pred[pred == frac] = torch.tensor(data["label_prompt"][i -
1]).to(pred.dtype)

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: mingxueg-nv

scripts/utils/trans_utils.py

@@ -349,7 +349,9 @@ def __call__(
                     pred += 0.5  # inplace mapping to avoid cloning pred
                     for i in range(1, object_num + 1):
                         frac = i + 0.5
-                        pred[pred == frac] = data["label_prompt"][i - 1].to(pred.dtype)
+                        pred[pred == frac] = torch.tensor(
+                            data["label_prompt"][i - 1]
+                        ).to(pred.dtype)
                     pred[pred == 0.5] = 0.0
                 data[keys] = pred
         return data