Formatting

yaph · Oct 16, 2024 · 0fc43c7 · 0fc43c7
1 parent bf0aabe
commit 0fc43c7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/bin/add_help_to_readme.py b/bin/add_help_to_readme.py
@@ -10,7 +10,7 @@
 quote = '```'
 
 f = io.StringIO()
-with (redirect_stdout(f), suppress(SystemExit)):
+with redirect_stdout(f), suppress(SystemExit):
     cli.main(['-h'])
 help_text = f.getvalue().replace('add_help_to_readme.py', 'charla')
 

diff --git a/charla/client/ollama.py b/charla/client/ollama.py
@@ -37,7 +37,8 @@ def generate(self, prompt: str):
         self.add_message(role='user', text=prompt)
 
         response = self.client.generate(
-            model=self.model, prompt=prompt, context=self.context, stream=True, system=self.system)
+            model=self.model, prompt=prompt, context=self.context, stream=True, system=self.system
+        )
 
         # Make sure system message is set only once.
         self.system = ''

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -8,4 +8,3 @@ def test_cli_help(capsys):
         main(['-h'])
     output = capsys.readouterr().out
     assert 'language model' in output
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -8,4 +8,3 @@ def test_cli_help(capsys):
		main(['-h'])
		output = capsys.readouterr().out
		assert 'language model' in output