Bugs fix and improvement for 0.1.8

mlchain/cli/init.py

@@ -9,7 +9,7 @@
 @click.argument('file', nargs=1, required=False, default='mlconfig.yaml')
 def init_command(file):
     if file is None:
-        file = 'mlconfig.yaml'
+        file = 'mlconfig.yaml'  # pragma: no cover
     if os.path.exists(file):
         logger.warning("File {} exists. Please change name file".format(file))
     else:

mlchain/cli/main.py

@@ -47,7 +47,7 @@ def main(as_module=False, is_testing=False):
     cli.add_command(serve_command)
     if is_testing:
         return cli
-    cli.main(args=sys.argv[1:], prog_name="python -m mlchain" if as_module else None)
+    cli.main(args=sys.argv[1:], prog_name="python -m mlchain" if as_module else None)  # pragma: no cover
 
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     cli = main(as_module=True)

mlchain/server/__init__.py

@@ -2,21 +2,18 @@
 
 try:
     from .flask_server import FlaskServer
-except Exception as ex:
+except Exception as ex:  # pragma: no cover
     import warnings
-
     warnings.warn("Can't import FlaskServer. {0}".format(ex))
 
 try:
     from .quart_server import QuartServer
-except Exception as ex:
+except Exception as ex:  # pragma: no cover
     import warnings
-
     warnings.warn("Can't import QuartServer. {0}".format(ex))
 
 try:
     from .grpc_server import GrpcServer
-except Exception as ex:
+except Exception as ex:  # pragma: no cover
     import warnings
-
     warnings.warn("Can't import GrpcServer. {0}".format(ex))