Fix profiler and benchmark

2023-04-17 21:28:24 -07:00 · 2023-04-17 21:28:24 -07:00 · 4546c4cffa
commit 4546c4cffa
parent 7b7ef39d0b
2 changed files with 2 additions and 2 deletions
--- a/benchmark.py
+++ b/benchmark.py
@ -25,7 +25,7 @@ print(
    round(
        1000000
        * timeit.timeit(
-            "gptc.compile(raw_model, max_ngram_length)",
+            "gptc.Model.compile(raw_model, max_ngram_length)",
            number=compile_iterations,
            globals=globals(),
        )
--- a/profiler.py
+++ b/profiler.py
@ -13,4 +13,4 @@ with open("models/raw.json") as f:
 with open("models/benchmark_text.txt") as f:
    text = f.read()

-cProfile.run("gptc.compile(raw_model, max_ngram_length)")
+cProfile.run("gptc.Model.compile(raw_model, max_ngram_length)")