Bump version to v0.2

google-deepmind · Nov 12, 2024 · 2ab965b · 2ab965b
1 parent 3a068a3
commit 2ab965b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "synthid-text"
-version = "0.1.1"
+version = "0.2"
 authors = [
   { name="DeepMind Technologies Limited" },
 ]

diff --git a/src/synthid_text/synthid_mixin_test.py b/src/synthid_text/synthid_mixin_test.py
@@ -31,7 +31,7 @@ class Config:
   is_encoder_decoder: bool = True
 
 
-class TestSynthIDModel(synthid_mixin.SynthIDSparseTopKMixin):
+class MockSynthIDModel(synthid_mixin.SynthIDSparseTopKMixin):
 
   def __init__(self, config: Config):
     super().__init__()
@@ -70,7 +70,7 @@ def test_sampling_from_mixin_includes_watermarking(self):
         autospec=True,
     ) as mock_watermarked_call:
       mock_watermarked_call.side_effect = old_watermarked_call
-      synthid_model = TestSynthIDModel(config=Config())
+      synthid_model = MockSynthIDModel(config=Config())
       generation_config = transformers.GenerationConfig(
           top_k=5,
           temperature=0.5,