Merge pull request #3 from kyegomez/master

Catching up 20240102 1012
kyegomez · Jan 2, 2024 · 6cda234 · 6cda234
2 parents a4f0547 + 1144fec
commit 6cda234
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/.github/workflows/generator-generic-ossf-slsa3-publish.yml b/.github/workflows/generator-generic-ossf-slsa3-publish.yml
@@ -60,7 +60,7 @@ jobs:
  actions: read # To read the workflow path.
  id-token: write # To sign the provenance.
  contents: write # To add assets to a release.
- uses: slsa-framework/slsa-github-generator/.github/workflows/generator_generic_slsa3.yml@v1.4.0
+ uses: slsa-framework/slsa-github-generator/.github/workflows/generator_generic_slsa3.yml@v1.9.0
  with:
  base64-subjects: "${{ needs.build.outputs.digests }}"
  upload-assets: true # Optional: Upload to a new release
diff --git a/.github/workflows/super-linter.yml b/.github/workflows/super-linter.yml
@@ -22,7 +22,7 @@ jobs:
  fetch-depth: 0
 
  - name: Lint Code Base
- uses: github/super-linter@v4
+ uses: github/super-linter@v5
  env:
  VALIDATE_ALL_CODEBASE: false
  DEFAULT_BRANCH: "master"

diff --git a/pyproject.toml b/pyproject.toml
@@ -29,7 +29,7 @@ transformers = "4.36.0"
 einops-exts = "0.0.4"
 torchvision = "*"
 accelerate = "0.25.0"
-datasets = "2.10.1"
+datasets = "2.16.1"
 lion-pytorch = "0.0.7"
 jax = "*"
 jaxlib = "*"

diff --git a/tests/structs/test_simple_vision_encoder.py b/tests/structs/test_simple_vision_encoder.py
@@ -1,9 +1,9 @@
 import torch
-from zeta.structs.simple_vision_encoder import SimpleVisionEncoder
+from zeta.structs.simple_vision_encoder import VisionEncoder
 
 
 def test_simple_vision_encoder_init():
- sve = SimpleVisionEncoder()
+ sve = VisionEncoder()
  assert sve.size == (384, 384)
  assert sve.model_name == "vikhyatk/moondream0"
  assert sve.return_shape is False
@@ -13,15 +13,15 @@ def test_simple_vision_encoder_init():
 
 
 def test_simple_vision_encoder_init_custom_size():
- sve = SimpleVisionEncoder(size=(512, 512))
+ sve = VisionEncoder(size=(512, 512))
  assert sve.size == (512, 512)
 
 
 def test_simple_vision_encoder_init_custom_model_name():
- sve = SimpleVisionEncoder(model_name="custom/model")
+ sve = VisionEncoder(model_name="custom/model")
  assert sve.model_name == "custom/model"
 
 
 def test_simple_vision_encoder_init_return_shape():
- sve = SimpleVisionEncoder(return_shape=True)
+ sve = VisionEncoder(return_shape=True)
  assert sve.return_shape is True