Import neuron package for Inferentia (#2157)

Author: RobertLucian

test/apis/batch/inferentia/handler.py

@@ -46,6 +46,7 @@ def __init__(self, config, job_spec):
             self.model.load_state_dict(torch.load(model_name))
             self.model.eval()
         elif config["device"] == "inf":
+            import torch_neuron
 
             self.model = torch.jit.load(model_name)
         else: