Here is the stack trace:
Traceback (most recent call last):
  File "test.py", line 131, in <module>
    profiler.run(input)
  File "test.py", line 28, in run
    return_val = super().run(*args)
  File "/home/keshav/anaconda3/envs/MolMPNN/lib/python3.7/site-packages/torch/fx/interpreter.py", line 116, in run
    self.env[node] = self.run_node(node)
  File "test.py", line 36, in run_node
    return_val = super().run_node(n)
  File "/home/keshav/anaconda3/envs/MolMPNN/lib/python3.7/site-packages/torch/fx/interpreter.py", line 142, in run_node
    return getattr(self, n.op)(n.target, args, kwargs)
  File "/home/keshav/anaconda3/envs/MolMPNN/lib/python3.7/site-packages/torch/fx/interpreter.py", line 169, in placeholder
    return next(self.args_iter)
StopIteration