Trajectree 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

trajectree/__init__.py

@@ -1,3 +0,0 @@
-import sys
-import os
-sys.path.append(os.path.abspath(__file__).replace('__init__.py', 'quimb'))

trajectree/fock_optics/devices.py

@@ -3,7 +3,7 @@ from scipy.linalg import expm
 import numpy as np
 from numpy import kron
 
-from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from quimb.tensor import MatrixProductOperator as mpo #type: ignore
 
 import qutip as qt
 

trajectree/fock_optics/light_sources.py

@@ -8,8 +8,8 @@ import numpy as np
 from numpy.linalg import matrix_power
 from numpy import kron, sqrt
 
-from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
 
 import qutip as qt
 from math import factorial

trajectree/fock_optics/measurement.py

@@ -9,9 +9,9 @@ import numpy as np
 from numpy.linalg import matrix_power
 from numpy import sqrt
 
-from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
-from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
 
 import qutip as qt
 from math import factorial
@@ -127,26 +127,26 @@ def bell_state_measurement(psi, N, site_tags, num_modes, efficiencies, dark_coun
     if return_MPOs:
         returned_MPOs = [U_BS_H, U_BS_V]
         if use_trajectory:
-            quantum_channel_list = [quantum_channel(N = N, num_modes = num_modes, formalism = "closed", unitary_MPOs = BSM_MPO, name = "BSM") for BSM_MPO in returned_MPOs]
+            quantum_channel_list = [quantum_channel(N = N, num_modes = num_modes, formalism = "closed", unitary_MPOs = BSM_MPO, name = "beam splitter") for BSM_MPO in returned_MPOs]
 
             damping_kraus_ops_0 = single_mode_bosonic_noise_channels(noise_parameter = 1-efficiencies[0], N = N)
             damping_kraus_ops_1 = single_mode_bosonic_noise_channels(noise_parameter = 1-efficiencies[1], N = N)
             two_mode_kraus_ops_0 = [sp.kron(op1, op2) for op1 in damping_kraus_ops_0 for op2 in damping_kraus_ops_0]
             two_mode_kraus_ops_1 = [sp.kron(op1, op2) for op1 in damping_kraus_ops_1 for op2 in damping_kraus_ops_1]
-            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops_0))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
-            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops_1))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops_0), name = "detector inefficiency")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops_1), name = "detector inefficiency")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
 
             amplification_kraus_ops_0 = single_mode_bosonic_noise_channels(noise_parameter = dark_counts_gain[0], N = N)
             amplification_kraus_ops_1 = single_mode_bosonic_noise_channels(noise_parameter = dark_counts_gain[1], N = N)
             two_mode_kraus_ops_0 = [sp.kron(op1, op2) for op1 in amplification_kraus_ops_0 for op2 in amplification_kraus_ops_0]
             two_mode_kraus_ops_1 = [sp.kron(op1, op2) for op1 in amplification_kraus_ops_1 for op2 in amplification_kraus_ops_1]
-            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops_0))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
-            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops_1))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops_0), name = "dark counts")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+            quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops_1), name = "dark counts")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
 
             BSM_POVM_1_OPs = generate_sqrt_POVM_MPO(sites=measurements[1], outcome = det_outcome, total_sites=num_modes, efficiency=1, N=N, pnr = pnr)
             BSM_POVM_1_OPs.extend(generate_sqrt_POVM_MPO(sites=measurements[0], outcome = 0, total_sites=num_modes, efficiency=1, N=N, pnr = pnr))
 
-            det_quantum_channels = [quantum_channel(N = N, num_modes = num_modes, formalism = "closed", unitary_MPOs = DET_MPO, name = "DET") for DET_MPO in BSM_POVM_1_OPs]
+            det_quantum_channels = [quantum_channel(N = N, num_modes = num_modes, formalism = "closed", unitary_MPOs = DET_MPO, name = "Det POVM") for DET_MPO in BSM_POVM_1_OPs]
             quantum_channel_list.extend(det_quantum_channels)
     
             return quantum_channel_list

trajectree/fock_optics/outputs.py

@@ -20,7 +20,7 @@ def generate_labels(num_systems, N):
         labels.append(new_label[:-1])
     return labels
 
-def read_quantum_state(TN_state, N, num_states = 4, return_dense = False, precision = 10):
+def read_quantum_state(TN_state, N, num_states = 4, return_dense = False, precision = 10, return_string = False):
     dense_state = TN_state.to_dense()
     if return_dense: return dense_state
     dense_state = np.reshape(dense_state.data, (-1, 1), order = 'C')
@@ -28,14 +28,18 @@ def read_quantum_state(TN_state, N, num_states = 4, return_dense = False, precis
     dense_state.data = np.round(dense_state.data, precision)
     dense_state.eliminate_zeros()
 
-    print_quantum_state(N, dense_state, num_states)
+    return print_quantum_state(N, dense_state, num_states, return_string)
 
-def print_quantum_state(N, dense_state, num_states = 4):
+def print_quantum_state(N, dense_state, num_states = 4, return_string = False):
     labels = generate_labels(num_states,N)
     state = dense_state.nonzero()[0]
-    print("Corresponding Basis terms:")
-    for k in state: print(labels[k],"-",k,"-",dense_state[k].data)
-
+    output = []
+    output.append("Corresponding Basis terms:")
+    for k in state: output.append(f"{labels[k]} - {k} - {dense_state[k].data}")
+    if not return_string:
+        print("\n".join(map(str, output)))
+    else:
+        return output
 
 
 def plot_coincidences(coincidence, idler_angles, signal_angles, title = ''):

trajectree/fock_optics/utils.py

@@ -1,12 +1,14 @@
 import numpy as np
 from numpy import sqrt
 
-from ..quimb.quimb.tensor import MatrixProductState as mps #type: ignore
-from ..quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
-from ..quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-from ..quimb.quimb.tensor.tensor_core import new_bond #type: ignore
-from ..quimb.quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
-from ..quimb.quimb.tensor.tensor_1d import TensorNetwork1DOperator #type: ignore
+from quimb.tensor import MatrixProductState as mps #type: ignore
+from quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from quimb.tensor.tensor_core import new_bond #type: ignore
+from quimb.tensor.tensor_1d_compress import enforce_1d_like #type: ignore
+from quimb.tensor.tensor_1d import TensorNetwork1DOperator #type: ignore
+
+from .outputs import read_quantum_state
 
 import qutip as qt
 import re
@@ -154,6 +156,7 @@ def create_bimode_bell_state(bell_state, N, error_tolerance = 1e-12):
     elif bell_state == "phi_minus":
         psi = tensor_network_apply_op_vec(NOT_MPO_0, vacuum, compress=True, contract = True, cutoff = error_tolerance)
 
+    # read_quantum_state(psi, N)
     
     psi = tensor_network_apply_op_vec(H_MPO, psi, compress=True, contract = True, cutoff = error_tolerance)
     # read_quantum_state(psi, N, num_states = 2)

trajectree/sequence/swap.py

@@ -2,6 +2,7 @@ from ..fock_optics.noise_models import *
 from ..fock_optics.measurement import *
 from ..fock_optics.utils import *
 from ..fock_optics.light_sources import *
+from ..fock_optics.outputs import *
 
 from ..trajectory import *
 
@@ -15,8 +16,8 @@ def generate_swapping_circuit(N, num_modes, site_tags, bsm_det_effs, bsm_dark_co
     # Amplitude damping due to fibers
     damping_kraus_ops = single_mode_bosonic_noise_channels(noise_parameter = channel_loss, N = N)
     two_mode_kraus_ops = [sp.kron(op, op) for op in damping_kraus_ops]
-    quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
-    quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops))) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+    quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((2,3), two_mode_kraus_ops), name = "fiber_attenuation")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
+    quantum_channel_list.append(quantum_channel(N = N, num_modes = num_modes, formalism = "kraus", kraus_ops_tuple = ((6,7), two_mode_kraus_ops), name = "fiber_attenuation")) # The tuples in this list are defined as (sites, kraus_ops). The sites are the sites where the Kraus ops are applied.
 
     # Quantum channel for the Bell state measurement
     # BSM_MPOs = bell_state_measurement(None, N, site_tags, num_modes, bsm_det_effs, error_tolerance, measurements = bsm_measurements, pnr = False, use_trajectory = True, return_MPOs = True, compress=True, contract=True)
@@ -50,7 +51,7 @@ def perform_swapping_simulation(N, num_modes, num_simulations, params, error_tol
     quantum_channels = generate_swapping_circuit(N, num_modes, psi.site_tags, [params["BSM_det_loss_1"], params["BSM_det_loss_2"]], [params["BSM_dark_counts_1"], params["BSM_dark_counts_2"]], params["BSM_meas"], params["channel_loss"], error_tolerance)
 
     if params["if_analyze_entanglement"]:
-        analyze_entanglement(quantum_channels, N, psi.site_tags, num_modes, params["PA_det_loss"], error_tolerance, params["alpha_list"], params["delta_list"])
+        analyze_entanglement(quantum_channels, N, psi.site_tags, num_modes, params["PA_det_eff"], error_tolerance, params["alpha_list"], params["delta_list"])
 
     t_eval = trajectory_evaluator(quantum_channels)
 
@@ -64,7 +65,7 @@ def perform_swapping_simulation(N, num_modes, num_simulations, params, error_tol
         probabilities.append(psi_iter.normalize())
         
         if params["calc_fidelity"]:
-            fidelity = np.abs(calc_fidelity_swapping(psi_iter, "psi_minus", N, error_tolerance))
+            fidelity = np.abs(calc_fidelity_swapping(psi_iter, "psi_plus", N, error_tolerance))
             fidelities.append(fidelity)
     
         time_taken = time.time() - start

trajectree/trajectory.py

@@ -1,7 +1,7 @@
 import numpy as np
-from .quimb.quimb.tensor import MatrixProductOperator as mpo #type: ignore
-from .quimb.quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
-
+from quimb.tensor import MatrixProductOperator as mpo #type: ignore
+from quimb.tensor.tensor_arbgeom import tensor_network_apply_op_vec #type: ignore
+from .fock_optics.outputs import read_quantum_state
 
 
 class quantum_channel:
@@ -84,7 +84,7 @@ class trajectory_evaluator():
     def cache_trajectree_node(self, trajectory_probs, trajectories):
         sorted_indices = np.argsort(trajectory_probs)
 
-        print("trajectory_probs", trajectory_probs)
+        # print("trajectory_probs", trajectory_probs)
 
         cached_trajectory_indices = sorted_indices[-self.cache_size:]
         cached_trajectories = np.array(trajectories)[cached_trajectory_indices]
@@ -206,5 +206,6 @@ class trajectory_evaluator():
             else:
                 # print("unitary skipped:", self.traversed_nodes)
                 pass
+            pass
 
         return psi

{trajectree-0.0.1.dist-info → trajectree-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Trajectree
-Version: 0.0.1
+Version: 0.0.3
 Summary: Trajectree is a quantum trajectory theory and tensor network based quantum optics simulator.
 Author-email: Ansh Singal <asingal@u.northwestern.edu>
 License-Expression: MIT
@@ -9,10 +9,9 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: numpy
-Requires-Dist: scipy
 Requires-Dist: matplotlib
 Requires-Dist: qutip
+Requires-Dist: trajectree-quimb
 Dynamic: license-file
 
 Trajectree is a quantum trajectory theory and tensor network based quantum optics simulator. 

trajectree-0.0.3.dist-info/RECORD

@@ -0,0 +1,16 @@
+trajectree/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+trajectree/optical_quant_info.py,sha256=27LWg0jIWT2ZXWVK8HEd-2gqAJl0GzmsyYi7sxyYzvA,7589
+trajectree/trajectory.py,sha256=mKyDZNDRyADouHL4Rf-JU3NYhEynVpUZmZgTl0EZLDE,11332
+trajectree/experimental/sparse.py,sha256=05aFsMcBSCUoBop1LJG_lB5kTULRSYhZhGT9RD-UH4Q,4851
+trajectree/fock_optics/devices.py,sha256=1gwl4AIktqu3obXmNhR1EoKR4iWulP4LfWD4xenKfx8,2130
+trajectree/fock_optics/light_sources.py,sha256=qzBevi1Uqcxb6KuYMuVQ9AkJbHCCigBf21N6s01x-V0,4823
+trajectree/fock_optics/measurement.py,sha256=n5syDcy8i_jYmPzAOV26wDg-Z98X0w_y5MmCBIYn3Gk,13222
+trajectree/fock_optics/noise_models.py,sha256=nyU_jNqjJkkcafq7vIXI8IRDiwthiOr4ZQqf55knZMM,1895
+trajectree/fock_optics/outputs.py,sha256=EMhs2gCwz6giLRyM09zxOLrpdbEuSG9axB4FFw6fuF4,2475
+trajectree/fock_optics/utils.py,sha256=iIz4DUluzpmW_sk1IlBeNYCiWZd6_eTi6I1p93DGt8U,7172
+trajectree/sequence/swap.py,sha256=MNc0E76icEWmaYRLOqXDj2ebEQwC2TlRftZwA8JBQkE,4378
+trajectree-0.0.3.dist-info/licenses/LICENSE,sha256=7EI8xVBu6h_7_JlVw-yPhhOZlpY9hP8wal7kHtqKT_E,1074
+trajectree-0.0.3.dist-info/METADATA,sha256=-zp8uGV6M2oHQTUvAcm_2FnXEgzxX6vg_U8bj0HYPMY,611
+trajectree-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+trajectree-0.0.3.dist-info/top_level.txt,sha256=6x9i8aAZcVn5tZ9J-2IVQMTGdn4bw6DiSd8pR3v4VR8,11
+trajectree-0.0.3.dist-info/RECORD,,

trajectree/quimb/docs/_pygments/_pygments_dark.py

@@ -1,118 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-Pygments version of the sublime Mariana theme.
-
-Pygments template by Jan T. Sott (https://github.com/idleberg)
-"""
-
-from pygments.style import Style
-from pygments.token import Keyword, Name, Comment, String, Error, Text, \
-     Number, Operator, Generic, Whitespace, Punctuation, Other, Literal
-
-
-BACKGROUND = "#1a1c1e"
-CURRENT_LINE = "#4e5a65"
-SELECTION = "#343d46"
-FOREGROUND = "#d8dee9"
-COMMENT = "#5a6272"
-RED = "#ec5f66"
-ORANGE = "#f9ae58"
-YELLOW = "#fac761"
-GREEN = "#99c794"
-AQUA = "#70c2bb"
-BLUE = "#6699cc"
-PURPLE = "#c695c6"
-
-
-class MarianaDark(Style):
-
-    default_style = ''
-
-    background_color = BACKGROUND
-    highlight_color = SELECTION
-
-    background_color = BACKGROUND
-    highlight_color = SELECTION
-
-    styles = {
-        # No corresponding class for the following:
-        Text:                      FOREGROUND,  # class:  ''
-        Whitespace:                "",          # class: 'w'
-        Error:                     RED,         # class: 'err'
-        Other:                     "",          # class 'x'
-
-        Comment:                   COMMENT,   # class: 'c'
-        Comment.Multiline:         "",        # class: 'cm'
-        Comment.Preproc:           "",        # class: 'cp'
-        Comment.Single:            "",        # class: 'c1'
-        Comment.Special:           "",        # class: 'cs'
-
-        Keyword:                   PURPLE,    # class: 'k'
-        Keyword.Constant:          RED,       # class: 'kc'
-        Keyword.Declaration:       "",        # class: 'kd'
-        Keyword.Namespace:         PURPLE,    # class: 'kn'
-        Keyword.Pseudo:            RED,       # class: 'kp'
-        Keyword.Reserved:          "",        # class: 'kr'
-        Keyword.Type:              YELLOW,    # class: 'kt'
-
-        Operator:                  RED,       # class: 'o'
-        Operator.Word:             "",        # class: 'ow' - like keywords
-
-        Punctuation:               AQUA,       # class: 'p'
-
-        Name:                      FOREGROUND,  # class: 'n'
-        Name.Attribute:            BLUE,        # class: 'na' - to be revised
-        Name.Builtin:              BLUE,          # class: 'nb'
-        Name.Builtin.Pseudo:       RED,         # class: 'bp'
-        Name.Class:                ORANGE,      # class: 'nc' - to be revised
-        Name.Constant:             RED,         # class: 'no' - to be revised
-        Name.Decorator:            AQUA,        # class: 'nd' - to be revised
-        Name.Entity:               BLUE,          # class: 'ni'
-        Name.Exception:            RED,         # class: 'ne'
-        Name.Function:             AQUA,        # class: 'nf'
-        Name.Function.Magic:       BLUE,        # class: 'nf'
-        Name.Property:             BLUE,        # class: 'py'
-        Name.Label:                BLUE,          # class: 'nl'
-        Name.Namespace:            BLUE,        # class: 'nn' - to be revised
-        Name.Other:                BLUE,        # class: 'nx'
-        Name.Tag:                  AQUA,        # class: 'nt' - like a keyword
-        Name.Variable:             RED,         # class: 'nv' - to be revised
-        Name.Variable.Class:       "",          # class: 'vc' - to be revised
-        Name.Variable.Global:      "",          # class: 'vg' - to be revised
-        Name.Variable.Instance:    "",          # class: 'vi' - to be revised
-
-        Number:                    ORANGE,    # class: 'm'
-        Number.Float:              "",        # class: 'mf'
-        Number.Hex:                "",        # class: 'mh'
-        Number.Integer:            "",        # class: 'mi'
-        Number.Integer.Long:       "",        # class: 'il'
-        Number.Oct:                "",        # class: 'mo'
-
-        Literal:                   ORANGE,    # class: 'l'
-        Literal.Date:              GREEN,     # class: 'ld'
-
-        String:                    GREEN,       # class: 's'
-        String.Backtick:           "",          # class: 'sb'
-        String.Char:               FOREGROUND,  # class: 'sc'
-        String.Doc:                COMMENT,     # class: 'sd' - like a comment
-        String.Double:             "",          # class: 's2'
-        String.Escape:             ORANGE,      # class: 'se'
-        String.Heredoc:            "",          # class: 'sh'
-        String.Interpol:           ORANGE,      # class: 'si'
-        String.Other:              "",          # class: 'sx'
-        String.Regex:              "",          # class: 'sr'
-        String.Single:             "",          # class: 's1'
-        String.Symbol:             "",          # class: 'ss'
-
-        Generic:                   "",                    # class: 'g'
-        Generic.Deleted:           RED,                   # class: 'gd',
-        Generic.Emph:              "italic",              # class: 'ge'
-        Generic.Error:             "",                    # class: 'gr'
-        Generic.Heading:           "bold " + FOREGROUND,  # class: 'gh'
-        Generic.Inserted:          GREEN,                 # class: 'gi'
-        Generic.Output:            "",                    # class: 'go'
-        Generic.Prompt:            "bold " + COMMENT,     # class: 'gp'
-        Generic.Strong:            "bold",                # class: 'gs'
-        Generic.Subheading:        "bold " + AQUA,        # class: 'gu'
-        Generic.Traceback:         "",                    # class: 'gt'
-    }

trajectree/quimb/docs/_pygments/_pygments_light.py

@@ -1,118 +0,0 @@
-# -*- coding: utf-8 -*-
-"""
-Pygments (light) version of the sublime Mariana theme.
-
-Pygments template by Jan T. Sott (https://github.com/idleberg)
-"""
-
-from pygments.style import Style
-from pygments.token import Keyword, Name, Comment, String, Error, Text, \
-     Number, Operator, Generic, Whitespace, Punctuation, Other, Literal
-
-
-BACKGROUND = "#f8f9fb"
-CURRENT_LINE = "#b7c0c8"
-SELECTION = "#d0d6dc"
-FOREGROUND = "#161c27"
-COMMENT = "#8d95a5"
-RED = "#e7323b"
-ORANGE = "#f79626"
-YELLOW = "#f9b52f"
-GREEN = "#70b069"
-AQUA = "#3d8f88"
-BLUE = "#407fbf"
-PURPLE = "#b474b4"
-
-
-class MarianaLight(Style):
-
-    default_style = ''
-
-    background_color = BACKGROUND
-    highlight_color = SELECTION
-
-    background_color = BACKGROUND
-    highlight_color = SELECTION
-
-    styles = {
-        # No corresponding class for the following:
-        Text:                      FOREGROUND,  # class:  ''
-        Whitespace:                "",          # class: 'w'
-        Error:                     RED,         # class: 'err'
-        Other:                     "",          # class 'x'
-
-        Comment:                   COMMENT,   # class: 'c'
-        Comment.Multiline:         "",        # class: 'cm'
-        Comment.Preproc:           "",        # class: 'cp'
-        Comment.Single:            "",        # class: 'c1'
-        Comment.Special:           "",        # class: 'cs'
-
-        Keyword:                   PURPLE,    # class: 'k'
-        Keyword.Constant:          RED,       # class: 'kc'
-        Keyword.Declaration:       "",        # class: 'kd'
-        Keyword.Namespace:         PURPLE,    # class: 'kn'
-        Keyword.Pseudo:            RED,       # class: 'kp'
-        Keyword.Reserved:          "",        # class: 'kr'
-        Keyword.Type:              YELLOW,    # class: 'kt'
-
-        Operator:                  RED,       # class: 'o'
-        Operator.Word:             "",        # class: 'ow' - like keywords
-
-        Punctuation:               AQUA,       # class: 'p'
-
-        Name:                      FOREGROUND,  # class: 'n'
-        Name.Attribute:            BLUE,        # class: 'na' - to be revised
-        Name.Builtin:              BLUE,          # class: 'nb'
-        Name.Builtin.Pseudo:       RED,         # class: 'bp'
-        Name.Class:                ORANGE,      # class: 'nc' - to be revised
-        Name.Constant:             RED,         # class: 'no' - to be revised
-        Name.Decorator:            AQUA,        # class: 'nd' - to be revised
-        Name.Entity:               BLUE,          # class: 'ni'
-        Name.Exception:            RED,         # class: 'ne'
-        Name.Function:             AQUA,        # class: 'nf'
-        Name.Function.Magic:       BLUE,        # class: 'nf'
-        Name.Property:             BLUE,        # class: 'py'
-        Name.Label:                BLUE,          # class: 'nl'
-        Name.Namespace:            BLUE,        # class: 'nn' - to be revised
-        Name.Other:                BLUE,        # class: 'nx'
-        Name.Tag:                  AQUA,        # class: 'nt' - like a keyword
-        Name.Variable:             RED,         # class: 'nv' - to be revised
-        Name.Variable.Class:       "",          # class: 'vc' - to be revised
-        Name.Variable.Global:      "",          # class: 'vg' - to be revised
-        Name.Variable.Instance:    "",          # class: 'vi' - to be revised
-
-        Number:                    ORANGE,    # class: 'm'
-        Number.Float:              "",        # class: 'mf'
-        Number.Hex:                "",        # class: 'mh'
-        Number.Integer:            "",        # class: 'mi'
-        Number.Integer.Long:       "",        # class: 'il'
-        Number.Oct:                "",        # class: 'mo'
-
-        Literal:                   ORANGE,    # class: 'l'
-        Literal.Date:              GREEN,     # class: 'ld'
-
-        String:                    GREEN,       # class: 's'
-        String.Backtick:           "",          # class: 'sb'
-        String.Char:               FOREGROUND,  # class: 'sc'
-        String.Doc:                COMMENT,     # class: 'sd' - like a comment
-        String.Double:             "",          # class: 's2'
-        String.Escape:             ORANGE,      # class: 'se'
-        String.Heredoc:            "",          # class: 'sh'
-        String.Interpol:           ORANGE,      # class: 'si'
-        String.Other:              "",          # class: 'sx'
-        String.Regex:              "",          # class: 'sr'
-        String.Single:             "",          # class: 's1'
-        String.Symbol:             "",          # class: 'ss'
-
-        Generic:                   "",                    # class: 'g'
-        Generic.Deleted:           RED,                   # class: 'gd',
-        Generic.Emph:              "italic",              # class: 'ge'
-        Generic.Error:             "",                    # class: 'gr'
-        Generic.Heading:           "bold " + FOREGROUND,  # class: 'gh'
-        Generic.Inserted:          GREEN,                 # class: 'gi'
-        Generic.Output:            "",                    # class: 'go'
-        Generic.Prompt:            "bold " + COMMENT,     # class: 'gp'
-        Generic.Strong:            "bold",                # class: 'gs'
-        Generic.Subheading:        "bold " + AQUA,        # class: 'gu'
-        Generic.Traceback:         "",                    # class: 'gt'
-    }

trajectree/quimb/docs/conf.py

@@ -1,158 +0,0 @@
-# Configuration file for the Sphinx documentation builder.
-#
-# For the full list of built-in configuration values, see the documentation:
-# https://www.sphinx-doc.org/en/master/usage/configuration.html
-
-import os
-import sys
-sys.path.append(os.path.abspath("./_pygments"))
-
-# -- Project information -----------------------------------------------------
-# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
-
-project = 'quimb'
-copyright = '2015-2024, Johnnie Gray'
-author = 'Johnnie Gray'
-
-# The full version, including alpha/beta/rc tags
-try:
-    from quimb import __version__
-    release = __version__
-except ImportError:
-    try:
-        from importlib.metadata import version as _version
-        release = _version('quimb')
-    except ImportError:
-        release = '0.0.0+unknown'
-
-version = '.'.join(release.split('.')[:2])
-
-# -- General configuration ---------------------------------------------------
-# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
-
-extensions = [
-    'sphinx.ext.intersphinx',
-    'sphinx.ext.extlinks',
-    'sphinx.ext.napoleon',
-    'sphinx.ext.linkcode',
-    'myst_nb',
-    "sphinx_design",
-    'sphinx_copybutton',
-    'autoapi.extension',
-]
-
-# msyt_nb configuration
-nb_execution_mode = "off"
-myst_heading_anchors = 4
-myst_enable_extensions = [
-    "amsmath",
-    "colon_fence",
-    "deflist",
-    "dollarmath",
-    "html_image",
-]
-
-
-# sphinx-autoapi
-autoapi_dirs = ['../quimb']
-
-extlinks = {
-    'issue': ('https://github.com/jcmgray/quimb/issues/%s', 'GH %s'),
-    'pull': ('https://github.com/jcmgray/quimb/pull/%s', 'PR %s'),
-}
-intersphinx_mapping = {
-    'python': ('https://docs.python.org/3/', None),
-    'numpy': ('https://numpy.org/doc/stable/', None),
-    'scipy': ('https://docs.scipy.org/doc/scipy/', None),
-    'cotengra': ('https://cotengra.readthedocs.io/en/latest/', None),
-    'autoray': ('https://autoray.readthedocs.io/en/latest/', None),
-}
-
-# Add any paths that contain templates here, relative to this directory.
-templates_path = ['_templates']
-
-# List of patterns, relative to source directory, that match files and
-# directories to ignore when looking for source files.
-# This pattern also affects html_static_path and html_extra_path.
-exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store', '**.ipynb_checkpoints']
-
-# -- Options for HTML output -------------------------------------------------
-# https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
-
-html_theme = 'furo'
-html_theme_options = {
-    "sidebar_hide_name": True,
-    "light_css_variables": {
-        # "color-brand-primary": "hsl(45, 80%, 45%)",
-        "color-brand-primary": "hsl(210, 50%, 50%)",
-        "color-brand-content": "hsl(210, 50%, 50%)",
-    },
-    "dark_css_variables": {
-        "color-brand-primary": "hsl(210, 50%, 60%)",
-        "color-brand-content": "hsl(210, 50%, 60%)",
-    },
-    "light_logo": "quimb_logo_title.png",
-    "dark_logo": "quimb_logo_title.png",
-}
-
-pygments_style = '_pygments_light.MarianaLight'
-pygments_dark_style = "_pygments_dark.MarianaDark"
-
-html_static_path = ['_static']
-html_css_files = ["my-styles.css"]
-html_favicon = "_static/quimb.ico"
-
-
-def linkcode_resolve(domain, info):
-    """
-    Determine the URL corresponding to Python object
-    """
-    import quimb
-    import inspect
-
-    if domain != "py":
-        return None
-
-    modname = info["module"]
-    fullname = info["fullname"]
-
-    submod = sys.modules.get(modname)
-    if submod is None:
-        return None
-
-    obj = submod
-    for part in fullname.split("."):
-        try:
-            obj = getattr(obj, part)
-        except AttributeError:
-            return None
-
-    try:
-        fn = inspect.getsourcefile(inspect.unwrap(obj))
-    except TypeError:
-        fn = None
-    if not fn:
-        return None
-
-    try:
-        source, lineno = inspect.getsourcelines(obj)
-    except OSError:
-        lineno = None
-
-    if lineno:
-        linespec = f"#L{lineno}-L{lineno + len(source) - 1}"
-    else:
-        linespec = ""
-
-    fn = os.path.relpath(fn, start=os.path.dirname(quimb.__file__))
-
-    if "+" in quimb.__version__:
-        return (
-            f"https://github.com/jcmgray/quimb/blob/"
-            f"main/quimb/{fn}{linespec}"
-        )
-    else:
-        return (
-            f"https://github.com/jcmgray/quimb/blob/"
-            f"v{quimb.__version__}/quimb/{fn}{linespec}"
-        )