Merge pull request #41 from asus4/fix-ort-importer

Fix ort importer
asus4 · Jan 25, 2025 · 017209a · 017209a
2 parents c6c3e49 + add1eac
commit 017209a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/com.github.asus4.onnxruntime.unity/Editor/OrtImporter.cs b/com.github.asus4.onnxruntime.unity/Editor/OrtImporter.cs
@@ -10,7 +10,11 @@ namespace Microsoft.ML.OnnxRuntime.Unity.Editor
     /// Sentis in installed and want to import the onnx model as OrtAsset,
     /// Select importer the dropdown of the onnx asset
     /// </summary>
+#if ORT_UNITY_SENTIS_ENABLED
     [ScriptedImporter(1, new[] { "ort" }, new[] { "onnx" })]
+#else
+    [ScriptedImporter(1, new[] { "ort", "onnx" })]
+#endif // ORT_UNITY_SENTIS_ENABLED
     public class OrtImporter : ScriptedImporter
     {
         public override void OnImportAsset(AssetImportContext ctx)

diff --git a/com.github.asus4.onnxruntime.unity/Editor/com.github.asus4.onnxruntime.unity.Editor.asmdef b/com.github.asus4.onnxruntime.unity/Editor/com.github.asus4.onnxruntime.unity.Editor.asmdef
@@ -13,6 +13,12 @@
     "precompiledReferences": [],
     "autoReferenced": true,
     "defineConstraints": [],
-    "versionDefines": [],
+    "versionDefines": [
+        {
+            "name": "com.unity.sentis",
+            "expression": "",
+            "define": "ORT_UNITY_SENTIS_ENABLED"
+        }
+    ],
     "noEngineReferences": false
 }