Fix minor issues

huggingface · Feb 15, 2024 · 230458f · 230458f
1 parent 39c9a02
commit 230458f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -14,3 +14,4 @@
 
 include README.md
 include LICENSE
+include optimum/neuron/utils/neuron_cc_wrapper
diff --git a/optimum/neuron/utils/hub_neuronx_cache.py b/optimum/neuron/utils/hub_neuronx_cache.py
@@ -309,6 +309,8 @@ def patch(restore: bool = False):
             patch()
         xm.rendezvous("Patch neuron_cc_wrapper")
         yield
+    except Exception as e:
+        raise e
     finally:
         if xm.get_ordinal() == 0:
             patch(restore=True)

diff --git a/optimum/neuron/utils/neuron_cc_wrapper b/optimum/neuron/utils/neuron_cc_wrapper
@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+##!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 import re
 import sys
Original file line number	Diff line number	Diff line change
Expand Up		@@ -14,3 +14,4 @@

		include README.md
		include LICENSE
		include optimum/neuron/utils/neuron_cc_wrapper