Skip to content

Commit

Permalink
flake8 evaluate.py
Browse files Browse the repository at this point in the history
  • Loading branch information
beckynevin committed Feb 14, 2024
1 parent 0c81d81 commit 9911d13
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions src/scripts/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ def predict(input, model):
"""
return 0


def load_inference_object(input_path):
"""
Expand All @@ -33,10 +34,17 @@ def load_inference_object(input_path):
return 0


if __name__ == '__main__':
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--checkpoint", type=str, help="Checkpoint to unloaded model checkpoint, either weights or the compressed model object")
parser.add_argument("--input", type=str, help="path to object to predict quality of")
parser.add_argument(
"--checkpoint",
type=str,
help="Checkpoint to unloaded model checkpoint,\
either weights or the compressed model object",
)
parser.add_argument(
"--input", type=str, help="path to object to predict quality of"
)
args = parser.parse_args()

model = load_model(args.checkpoint)
Expand Down

0 comments on commit 9911d13

Please sign in to comment.