Minor changes for tests.

.github/workflows/unit_testing.yml

@@ -33,5 +33,6 @@ jobs:
         shell: bash
       - name: Test with pytest
         run: |
-          python -m unittest discover ./tests/ 'test*.py'
+          cd tests
+          python -m unittest discover . 'test*.py'
         shell: bash

graphmuse/loader/neighbor_loader.py

@@ -143,6 +143,8 @@ def sample_hetero_graph(self, data, target_nodes, colptr_dict, row_dict, to_edge
                 self.num_neighbors.get_mapped_values(data.edge_types),
                 self.node_time,
             )
+            if torch_geometric.typing.WITH_EDGE_TIME_NEIGHBOR_SAMPLE:
+                args += (self.edge_time,)
             # Setting to None
             seed_time = None
             args += (seed_time,)

setup.py

@@ -33,7 +33,7 @@
 
 ext_modules = [
     setuptools.Extension(
-        name="graphmuse.samplers.csamplers", sources=[os.path.join("src", "gmsamplersmodule.c")], extra_compile_args = eca,
+        name="graphmuse.samplers.csamplers", sources=[os.path.join(os.path.dirname(__file__), "src", "gmsamplersmodule.c")],
             extra_link_args = [], include_dirs=[os.path.join(numpy.get_include(), "numpy"), "include"])]
 
 # os.environ["CC"] = "gcc"
@@ -42,7 +42,7 @@
 long_description = open(os.path.join(os.path.dirname(__file__), 'README.md'), "r").read()
 
 setuptools.setup(
-    name='GraphMuse',
+    name='graphmuse',
     version='0.0.1',
     description='GraphMuse is a Python Library for Graph Deep Learning on Symbolic Music.',
     long_description=long_description,