Commit d895bfe6 authored by Stephen Hogg's avatar Stephen Hogg
Browse files

Update dummy model classmethod args

parent 05590e11
...@@ -7,7 +7,7 @@ class DummyLM(LM): ...@@ -7,7 +7,7 @@ class DummyLM(LM):
pass pass
@classmethod @classmethod
def create_from_arg_string(cls, arg_string): def create_from_arg_string(cls, arg_string, additional_config=None):
return cls() return cls()
def loglikelihood(self, requests): def loglikelihood(self, requests):
...@@ -23,7 +23,7 @@ class DummyLM(LM): ...@@ -23,7 +23,7 @@ class DummyLM(LM):
for ctx, _ in requests: for ctx, _ in requests:
res.append("lol") res.append("lol")
assert ctx.strip() != '' assert ctx.strip() != ""
return res return res
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment