diff --git a/samples/code_execution.py b/samples/code_execution.py index ebc44b828..dfc2ce84e 100644 --- a/samples/code_execution.py +++ b/samples/code_execution.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_code_execution_basic(self): # [START code_execution_basic] diff --git a/samples/configure_model_parameters.py b/samples/configure_model_parameters.py index dbc33dd08..3fe28a5a8 100644 --- a/samples/configure_model_parameters.py +++ b/samples/configure_model_parameters.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_configure_model(self): # [START configure_model_parameters] diff --git a/samples/embed.py b/samples/embed.py index 33f46690b..a5897639b 100644 --- a/samples/embed.py +++ b/samples/embed.py @@ -15,14 +15,11 @@ from absl.testing import absltest - - class UnitTests(absltest.TestCase): def test_embed_content(self): # [START embed_content] import google.generativeai as genai - text = "Hello World!" result = genai.embed_content( model="models/text-embedding-004", content=text, output_dimensionality=10 diff --git a/samples/function_calling.py b/samples/function_calling.py index 35ad45100..829b97742 100644 --- a/samples/function_calling.py +++ b/samples/function_calling.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_function_calling(self): # [START function_calling] diff --git a/samples/models.py b/samples/models.py index 97018a887..1eea4b31e 100644 --- a/samples/models.py +++ b/samples/models.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_models_list(self): # [START models_list] diff --git a/samples/safety_settings.py b/samples/safety_settings.py index 679d27bc1..49b099dad 100644 --- a/samples/safety_settings.py +++ b/samples/safety_settings.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_safety_settings(self): # [START safety_settings] diff --git a/samples/system_instruction.py b/samples/system_instruction.py index e827d7d75..bf486ec45 100644 --- a/samples/system_instruction.py +++ b/samples/system_instruction.py @@ -15,7 +15,6 @@ from absl.testing import absltest - class UnitTests(absltest.TestCase): def test_system_instructions(self): # [START system_instruction]