forgotten dict

pymc-devs · Oct 27, 2022 · b3ca27d · b3ca27d
1 parent 117b929
commit b3ca27d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc/aesaraf.py b/pymc/aesaraf.py
@@ -556,7 +556,7 @@ def join_nonshared_inputs(
         by a joined vector input.
     inputs: list of TensorVariable
         List of input TensorVariables which will be replaced by a joined vector input.
-    shared_inputs: dict of TensorVariable to TensorSharedVariable, optional
+    shared_inputs: dict of {TensorVariable : TensorSharedVariable}, optional
         Dict of TensorVariable and their associated TensorSharedVariable in
         subgraph replacement.
     make_inputs_shared: bool, defaults to False