Skip to content

Commit

Permalink
quick version tick and re-release
Browse files Browse the repository at this point in the history
  • Loading branch information
joel99 committed Apr 25, 2024
1 parent f5db887 commit ba5c304
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion decoder_demos/sklearn_sample.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ ADD ./preproc/filtering.py filtering.py
ENV SPLIT "h1"
# ENV PHASE "minival"
ENV PHASE "test"
ENV BATCH_SIZE 8

# Make sure this matches the mounted data volume path. Generally leave as is.
ENV EVAL_DATA_PATH "/dataset/evaluation_data"
Expand All @@ -49,4 +50,4 @@ RUN pwd
# CMD specifies a default command to run when the container is launched.
# It can be overridden with any cmd e.g. sudo docker run -it my_image /bin/bash
CMD ["/bin/bash", "-c", \
"python decode.py --evaluation $EVALUATION_LOC --model-path data/decoder.pkl --split $SPLIT --phase $PHASE"]
"python decode.py --evaluation $EVALUATION_LOC --model-path data/decoder.pkl --split $SPLIT --phase $PHASE --batch-size $BATCH_SIZE"]
1 change: 1 addition & 0 deletions falcon_challenge/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,7 @@
FalconTask.h1: 8,
FalconTask.m1: 4,
FalconTask.h2: 1,
FalconTask.m2: 1, # TODO up
}

# Development time flag. False allows direct evaluation without payload writing, only usable for local minival.
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

setup(
name='falcon_challenge',
version='0.2.8',
version='0.2.9',

url='https://github.com/snel-repo/stability-benchmark',
author='Joel Ye',
Expand Down

0 comments on commit ba5c304

Please sign in to comment.