Насколько я знаю, не существует специального способа передачи аргументов командной строки в код python. Это пример рабочего кода, который я использую при создании tfrecords.
!python generate_tfrecord.py --csv_input=data/test_labels.csv --output_path=data/test.record --image_dir=images/
Я не вижу никакой разницы между передачей аргументов обычной командной строки python и colab. Пожалуйста, добавьте больше кода к своему вопросу, чтобы получить лучшую помощь.