isinstance(value, Dict[str, torch.Tensor]) does not work

[ailzhang]

import torch
from typing import Dict, Any
@torch.jit.ignore
def dbg(x: Any) -> None:
  print(type(x))
  print(x)
  print(isinstance(x, dict))
  # this doesn't work in python actually: Parameterized generics cannot be used with class or instance checks
  # print(isinstance(x, Dict[str, torch.Tensor]))

@torch.jit.script
def foo(x : Any):
  # dbg(x)
  assert isinstance(x, Dict[str, torch.Tensor])
  return x["a"]

foo({'a': torch.rand(5)})

cc @gmagogsfm

[gmagogsfm]

This is being worked on in #37516.

[eellison]

@gmagogsfm I think this is a different issue. Here we're correctly generating the right IR something is just borking in the runtime

[Lilyjjo]

This code snippet runs now, I'm going to close the issue.