From 4546c4cffaff0f2fc94b0dd446cf1be85d1fee1e Mon Sep 17 00:00:00 2001
From: Samuel Sloniker <sam@kj7rrv.com>
Date: Mon, 17 Apr 2023 21:28:24 -0700
Subject: [PATCH] Fix profiler and benchmark

---
 benchmark.py | 2 +-
 profiler.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/benchmark.py b/benchmark.py
index 00834b0..84106f8 100644
--- 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(),
         )
diff --git a/profiler.py b/profiler.py
index b581951..4891541 100644
--- 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)")