fix: add missing param

runpod-workers · Dec 20, 2023 · 10b177b · 10b177b
1 parent f854051
commit 10b177b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/CI-test_worker.yml b/.github/workflows/CI-test_worker.yml
@@ -50,7 +50,7 @@ jobs:
  run: |
  python -m pip install --upgrade pip
  pip install -r builder/requirements.txt
- apt-get install ffmpeg libsm6 libxext6
+ apt-get install ffmpeg libsm6 libxext6 -y
 
  - name: Execute Tests
  run: |

diff --git a/src/rp_schemas.py b/src/rp_schemas.py
@@ -58,5 +58,10 @@
  'required': False,
  'default': 1,
  'constraints': lambda img_count: 3 > img_count > 0
- }
+ },
+ 'high_noise_frac': {
+ 'type': float,
+ 'required': False,
+ 'default': None
+ },
 }