add tests dependencies to workflow

Tauffer-Consulting · Oct 23, 2023 · 33f3b1c · 33f3b1c
1 parent 7ce29e5
commit 33f3b1c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 9 deletions.
diff --git a/.github/workflows/validate-and-organize.yml b/.github/workflows/validate-and-organize.yml
@@ -42,7 +42,7 @@ jobs:
           domino piece organize --build-images --source-url=https://github.com/${{github.repository}}
 
       - name: Install Tests Dependencies
-        run: pip install pytest
+        run: pip install pytest && pip install tiktoken
 
       - name: Run tests over built images
         env:

diff --git a/config.toml b/config.toml
@@ -10,4 +10,4 @@ REPOSITORY_LABEL = "OpenAI Domino Pieces"
 # The version of this Pieces release
 # Attention: changing this will create a new release 
 VERSION = "0.4.0"
-
+  
diff --git a/pieces/TextSummarizerPiece/test_text_summarizer_piece.py b/pieces/TextSummarizerPiece/test_text_summarizer_piece.py
@@ -68,10 +68,3 @@ def test_piece():
     encoding = tiktoken.encoding_for_model(piece_kwargs["openai_model"])
     text_tokens = encoding.encode(text=generated_prompt)
     assert len(text_tokens) <= piece_kwargs["completion_max_tokens"]
-
-
-
-
-
-if __name__ == "__main__":
-    test_piece()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -10,4 +10,4 @@ REPOSITORY_LABEL = "OpenAI Domino Pieces"
		# The version of this Pieces release
		# Attention: changing this will create a new release
		VERSION = "0.4.0"