Merge pull request #76 from eecs485staff/bugfix-version-flag

Get version with importlib.metadata
eecs485staff · Mar 29, 2024 · b9356dd · b9356dd
2 parents 63a8f6e + ff650bc
commit b9356dd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/madoop/__main__.py b/madoop/__main__.py
@@ -4,12 +4,12 @@
 
 """
 import argparse
+import importlib.metadata
 import logging
 import pathlib
 import shutil
 import sys
 import textwrap
-import pkg_resources
 from .mapreduce import mapreduce
 from .exceptions import MadoopError
 
@@ -21,10 +21,10 @@ def main():
     )
 
     optional_args = parser.add_argument_group('optional arguments')
-    version = pkg_resources.get_distribution("madoop").version
+
     optional_args.add_argument(
         '--version', action='version',
-        version=f'Madoop {version}'
+        version=f'Madoop {importlib.metadata.version("madoop")}'
     )
     optional_args.add_argument(
         '--example', action=ExampleAction, nargs=0,