fixed comments

tardis/montecarlo/base.py

@@ -94,6 +94,8 @@ def __init__(
         self.use_gpu = use_gpu
 
         self.virt_logging = enable_virtual_packet_logging
+
+        # Length 2 for initialization - will be removed in next PR
         self.virt_packet_last_interaction_type = np.ones(2) * -1
         self.virt_packet_last_interaction_in_nu = np.ones(2) * -1.0
         self.virt_packet_last_line_interaction_in_id = np.ones(2) * -1

tardis/montecarlo/montecarlo_numba/tests/test_cuda_formal_integral.py

@@ -1,28 +1,17 @@
-import pytest
 import numpy as np
-from tardis import constants as c
-from astropy import units as u
-
-from copy import deepcopy
 import numpy.testing as ntest
+import pytest
 from numba import cuda
-from numba import njit
-from tardis.montecarlo.montecarlo_numba import njit_dict_no_parallel
 
-
-import tardis.montecarlo.montecarlo_numba.formal_integral_cuda as formal_integral_cuda
 import tardis.montecarlo.montecarlo_numba.formal_integral as formal_integral_numba
+import tardis.montecarlo.montecarlo_numba.formal_integral_cuda as formal_integral_cuda
+from tardis import constants as c
 from tardis.model.geometry.radial1d import NumbaRadial1DGeometry
-from tardis.montecarlo.montecarlo_numba.numba_interface import NumbaModel
-
-
-from tardis.montecarlo.montecarlo_numba.formal_integral import FormalIntegrator
 from tardis.montecarlo.montecarlo_numba.formal_integral import (
+    FormalIntegrator,
     NumbaFormalIntegrator,
 )
-
-from tardis.montecarlo.base import MontecarloTransportSolver
-
+from tardis.montecarlo.montecarlo_numba.numba_interface import NumbaModel
 
 # Test cases must also take into account use of a GPU to run. If there is no GPU then the test cases will fail.
 GPUs_available = cuda.is_available()