badaoui commited on
Commit
c9dede9
·
1 Parent(s): a4a38a2
Files changed (1) hide show
  1. optimum_neuron_export.py +2 -2
optimum_neuron_export.py CHANGED
@@ -173,13 +173,13 @@ Neuron-optimized models can achieve high-performance inference on AWS Inferentia
173
  if os.path.exists(folder):
174
  shutil.rmtree(folder, ignore_errors=True)
175
  print(f"Conversion failed with error: {e}") # Print the actual error
176
- return str(e) # Return error code and message
177
 
178
  finally:
179
  # Ensure cleanup
180
  if os.path.exists(folder):
181
  shutil.rmtree(folder, ignore_errors=True)
182
 
183
- return new_pr
184
 
185
 
 
173
  if os.path.exists(folder):
174
  shutil.rmtree(folder, ignore_errors=True)
175
  print(f"Conversion failed with error: {e}") # Print the actual error
176
+ return "1", str(e) # Return error code and message
177
 
178
  finally:
179
  # Ensure cleanup
180
  if os.path.exists(folder):
181
  shutil.rmtree(folder, ignore_errors=True)
182
 
183
+ return "0", new_pr
184
 
185