Fix module handling when installing as a package

prepare_pyspark_venv.py

@@ -10,7 +10,7 @@ def setup_env(venv_name, req_file):
     virtualenvs_folder = os.path.expanduser("~/.virtualenvs")
     venv_dir = os.path.join(virtualenvs_folder, venv_name)
     virtualenv.cli_run([venv_dir])
-    cmd = f". {virtualenvs_folder}/{venv_name}/bin/activate && pip install -r {req_file} && pip install {SCRIPT_PATH}"
+    cmd = f". {virtualenvs_folder}/{venv_name}/bin/activate && pip3 install -r {req_file} && pip3 install {SCRIPT_PATH}"
     os.system(cmd)
 
 

setup.py

@@ -1,16 +1,22 @@
-from setuptools import find_packages, setup
-
-setup(
-   name='osm_address',
-   version='1.0',
-   description='Module to handle OSM data for address extraction',
-   author='Björn Wächter',
-   author_email='[email protected]',
-   packages=find_packages(include=['osm_address', 'osm_address.*']),
-   install_requires=[
-      'apache-sedona>=1.5.0',
-      'pyarrow==14.0.1',
-      'pyspark==3.3.1',
-      'shapely==2.0.2'
-   ]
-)
+from setuptools import find_packages, setup
+
+setup(
+   name='osm_address',
+   version='1.0',
+   description='Module to handle OSM data for address extraction',
+   author='Björn Wächter',
+   author_email='[email protected]',
+   packages=find_packages(include=[
+      'osm_address',
+      'osm_address.osm',
+      'osm_address.transform',
+      'osm_address.udf',
+      'osm_address.utils'
+   ]),
+   install_requires=[
+      'apache-sedona>=1.5.0',
+      'pyarrow==14.0.1',
+      'pyspark==3.3.1',
+      'shapely==2.0.2'
+   ]
+)