Skip to content

Commit

Permalink
correct format
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosuc3m committed Nov 13, 2024
1 parent 1c43812 commit abbf369
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/create_macro.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import os
import argparse

MACRO_STR = "run(\"DeepImageJ Run\", \"modelPath={modelPath} inputPath={inputPath} outputFolder={outputFolder} displayOutput=null\")"
MACRO_STR = "run(\"DeepImageJ Run\", \"modelPath={model_path} inputPath={input_path} outputFolder={output_folder} displayOutput=null\")"

# Create the argument parser
parser = argparse.ArgumentParser()
Expand Down Expand Up @@ -73,6 +73,6 @@ def parse_model_paths(models):
print(mfp)
descriptor = ModelDescriptorFactory.readFromLocalFile(os.path.join(mfp, Constants.RDF_FNAME))
sample_name = descriptor.getInputTensors().get(0).getSampleTensorName()
macro = MACRO_STR.format(model_path=mfp, input_path=os.path.join(mfp, sample_name), outputFolder=mfp)
macro = MACRO_STR.format(model_path=mfp, input_path=os.path.join(mfp, sample_name), output_folder=mfp)
file.write(macro + os.linesep)

0 comments on commit abbf369

Please sign in to comment.