python3Packages.executorch: fix failing test

2026-06-11 07:41:18 +00:00
parent e4937ac558
commit 45f0137e81
1 changed files with 7 additions and 0 deletions
@@ -102,6 +102,13 @@ buildPythonPackage.override { inherit (torch) stdenv; } (finalAttrs: {

      sed -i "1i #include <cstdint>" backends/apple/coreml/runtime/inmemoryfs/memory_buffer.hpp
      sed -i "1i #include <cstdint>" extension/llm/tokenizers/third-party/sentencepiece/src/sentencepiece_processor.h
+    ''
+    # AssertionError: '1.3.0' != '0.1.0'
+    + ''
+      substituteInPlace extension/llm/tokenizers/test/test_python_bindings.py \
+        --replace-fail \
+          'self.assertEqual(pytorch_tokenizers.__version__, "0.1.0")' \
+          'self.assertEqual(pytorch_tokenizers.__version__, "${pytorch-tokenizers.version}")'
    '';

  env = {