diff --git a/main.py b/main.py index f155eb3..05e4bf7 100644 --- a/main.py +++ b/main.py @@ -95,12 +95,12 @@ if __name__ == "__main__": args = parser.parse_args() sys.stderr = open(osp.join(osp.dirname(__file__),'stderr.log'),'w') args.arg = ' '.join(args.arg) - + if not args.arg.isdecimal(): - # save on import time slightly - from print2d import print2d - if any(arg for arg in (args.route,args.direction)): - from search import Search,StopSearch + # save on import time slightly + from print2d import print2d + if any(arg for arg in (args.route,args.direction)): + from search import Search,StopSearch #routes if not args.route: data = ctabus.get_routes()['routes']