How to call a python script with parsing arguments on jupyter notebook

Question

I have a python script that I defined:

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--task', type=str)
    parser.add_argument('--scale', type=int)
    args = parser.parse_args()
    ... # Do things with my arguments

if __name__ == '__main__':
        main()

And I call this script on command line doing:

python myscript.py --task mytask --scale 1

I would like to call this script in a jupyter notebook. Is there a way to do this parsing the arguments and not modifying my script at all? I.e., doing something that looks like to this:

import myscript    
myscript.main(--task=mytask,scale=1)

P.S.: I tried using magic line such as %run (that probably could work in my case as well) but I had trouble collecting the returns of my script.

Chris · Accepted Answer · 2022-03-16 08:45:13Z

1

You can pass args to parser.parse_args:

# myscript.py
import argparse

def main(args=None):
    parser = argparse.ArgumentParser()
    parser.add_argument('--task', type=str)
    parser.add_argument('--scale', type=int)
    args = parser.parse_args(args=args)
    print("task is: ", args.task)
    print("scale is: ", args.scale)

if __name__ == "__main__":
    main()

Output from cli:

python3 myscript.py --task aaa --scale 10
# task is:  aaa
# scale is:  10

Output from python

import myscript

myscript.main("--task aaa --scale 10".split())

# task is:  aaa
# scale is:  10

answered Mar 16, 2022 at 8:45

Chris

29.8k3 gold badges34 silver badges56 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

brenodacosta Over a year ago

I was thinking if it would be possible do something letting def main():. But this clearly solves my problem completely.

Wayne Over a year ago

OP mentioned %run in the 'P.S.' of the post. This script would work fine in a notebook being called with %run myscript.py --task mytask --scale 1 as well. Adding this here more for others looking how to generally run a Python script in a notebook given the title of this post.

Collectives™ on Stack Overflow

How to call a python script with parsing arguments on jupyter notebook

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related