Publish pyfuncserver executable

python/pyfunc-server/pyfuncserver/__main__.py

@@ -18,21 +18,26 @@
 import traceback
 
 import uvloop
-
 from pyfuncserver.config import Config
 from pyfuncserver.model.model import PyFuncModel
 from pyfuncserver.server import PyFuncServer
 from pyfuncserver.utils.contants import ERR_DRY_RUN
 
 parser = argparse.ArgumentParser()
-parser.add_argument('--model_dir', required=True,
-                    help='A URI pointer to the model binary')
-parser.add_argument('--dry_run', default=False, action='store_true', required=False,
-                    help="Dry run pyfunc server by loading the specified model "
-                         "in --model_dir without starting webserver")
+parser.add_argument(
+    "--model_dir", required=True, help="A URI pointer to the model binary"
+)
+parser.add_argument(
+    "--dry_run",
+    default=False,
+    action="store_true",
+    required=False,
+    help="Dry run pyfunc server by loading the specified model "
+    "in --model_dir without starting webserver",
+)
 args, _ = parser.parse_known_args()
 
-logging.getLogger('tornado.access').disabled = True
+logging.getLogger("tornado.access").disabled = True
 
 if __name__ == "__main__":
     # use uvloop as the event loop

python/pyfunc-server/pyfuncserver/version.py

@@ -0,0 +1 @@
+VERSION = "0.40.0"

python/pyfunc-server/setup.py

@@ -12,10 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import imp
 import os
 
 from setuptools import find_packages, setup
 
+version = imp.load_source(
+    "pyfuncserver.version", os.path.join("pyfuncserver", "version.py")
+).VERSION
+
 tests_require = [
     "joblib>=0.13.0,<1.2.0",  # >=1.2.0 upon upgrade of kserve's version
     "mypy",
@@ -40,13 +45,13 @@
 merlin_sdk_package = "merlin-sdk"
 for index, item in enumerate(REQUIRE):
     if merlin_sdk_package in item:
-        REQUIRE[
-            index
-        ] = f"{merlin_sdk_package} @ file://localhost/{merlin_path}#egg={merlin_sdk_package}"
+        REQUIRE[index] = (
+            f"{merlin_sdk_package} @ file://localhost/{merlin_path}#egg={merlin_sdk_package}"
+        )
 
 setup(
     name="pyfuncserver",
-    version="0.6.0",
+    version=version,
     author_email="[email protected]",
     description="Model Server implementation for mlflow pyfunc model",
     long_description=open("README.md").read(),
@@ -56,4 +61,8 @@
     install_requires=REQUIRE,
     tests_require=tests_require,
     extras_require={"test": tests_require},
+    entry_points="""
+        [console_scripts]
+        pyfuncserver=pyfuncserver.__main__:main
+    """,
 )