feat: add ability to do vector-vector loop

.github/workflows/test_suite.yaml

@@ -45,27 +45,26 @@ jobs:
         run: |
           python -m pytest --ignore tests/test_plot.py --cov=matvis --cov-config=.coveragerc --cov-report xml:./coverage.xml --durations=25 --log-cli-level=${{ env.LOG_LEVEL}}
 
-      - name: Run Plotting Tests
-        if: matrix.python-version != '3.9' || matrix.os == 'macos-latest'
-        run: |
-          python -m pytest tests/test_plot.py --cov=matvis --cov-config=.coveragerc --cov-append --cov-report xml:./coverage.xml --durations=25 --log-cli-level=INFO
-
       - name: Upload coverage report
-        uses: codecov/codecov-action@v3.1.3
+        uses: codecov/codecov-action@v4
         with:
           file: ./coverage.xml
           flags: unittests
           name: codecov-umbrella
           fail_ci_if_error: true
+          token: ${{ secrets.CODECOV_TOKEN }}
+
 
   gpu_tests:
     env:
       ENV_NAME: tests
-      PYTHON: "3.10"
+      PYTHON: "3.12"
       OS: Ubuntu
     name: Self-Hosted Tests (GPU)
     runs-on: [self-hosted, gpu]
-
+    defaults:
+      run:
+        shell: bash -el {0}
     steps:
       - name: Add Home to PATH
         run: |
@@ -76,28 +75,30 @@ jobs:
         with:
           node-version: 16
 
-      - uses: actions/checkout@v3.6.0
+      - uses: actions/checkout@v4
         with:
           fetch-depth: 0
 
-      - name: Setup Python
-        uses: actions/setup-python@v4
-        with:
-          python-version: ${{ env.PYTHON }}
-
       - name: Install
         run: |
+          echo $(whoami)
+          source ~/.bashrc
+          micromamba activate testenv
           echo $(which pip)
-          pip install .[test,gpu]
+          python -m pip install --upgrade pip
+          python -m pip install .[test,gpu]
 
       - name: Run Tests
         run: |
+          source ~/.bashrc
+          micromamba activate testenv
           python -m pytest -k "gpu" --cov=matvis --cov-config=.coveragerc --cov-report xml:./coverage.xml --durations=25 --log-cli-level=INFO
 
       - name: Upload coverage report
-        uses: codecov/codecov-action@v3.1.3
+        uses: codecov/codecov-action@v4
         with:
           file: ./coverage.xml
           flags: unittests
           name: codecov-umbrella
           fail_ci_if_error: true
+          token: ${{ secrets.CODECOV_TOKEN }}

.pre-commit-config.yaml

@@ -32,7 +32,7 @@ repos:
       - flake8-print
 
 - repo: https://github.com/psf/black-pre-commit-mirror
-  rev: 24.4.2
+  rev: 24.8.0
   hooks:
   - id: black
 

ci/test-env.yml

@@ -1,4 +1,4 @@
-name: tests
+name: testenv
 channels:
   - conda-forge
   - defaults
@@ -15,6 +15,7 @@ dependencies:
   - h5py>=3.2
   - ffmpeg
   - pyuvdata  # For testing when using UVBeam object
+  - cupy
   - pip:
     - pyuvsim[sim]>=1.2
     - pyradiosky>=0.1.1

setup.cfg

@@ -20,9 +20,13 @@ classifiers =
 packages = find_namespace:
 install_requires =
     astropy
-    numpy
+    click
+    docstring-parser
+    line-profiler
+    numpy>=2.0
     psutil
-    pyuvdata>=2.2.8
+    pyuvdata@git+https://github.com/RadioAstronomySoftwareGroup/pyuvdata
+    rich
     scipy
 python_requires = >=3.9
 include_package_data = True
@@ -51,20 +55,21 @@ docs =
     numpydoc
     sphinx
 gpu =
+    cupy
     jinja2
-    pycuda
-    scikit-cuda
 profile =
     click
     line-profiler
     pyuvsim>=1.2.5
 test =
+    astropy-healpix
     hypothesis
     ipython
     matplotlib
     pyradiosky
     pytest
     pytest-cov
+    pytest-lazy-fixtures
     pyuvsim[sim]>=1.2.5
 
 [test]
@@ -107,6 +112,7 @@ ignore =
     D401
     B007
     G004
+    D107
 per-file-ignores =
     src/matvis/cli.py: T201,T001
     tests/*.py: T201,T001,E402

src/matvis/__init__.py

@@ -14,16 +14,19 @@
     del get_distribution, DistributionNotFound
 
 try:
-    import pycuda
+    import cupy
 
     HAVE_GPU = True
 except ImportError:
     HAVE_GPU = False
 
 
-from . import cpu, gpu, plot
+from . import cpu
 from .wrapper import simulate_vis
 
+if HAVE_GPU:
+    from . import gpu
+
 DATA_PATH = Path(__file__).parent / "data"
 
 del Path