fixed docstring typo

jakobrunge · Oct 11, 2023 · 5413459 · 5413459
1 parent 50a407c
commit 5413459
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 39 deletions.
diff --git a/setup.py b/setup.py
@@ -60,7 +60,7 @@ def run(self):
 # Run the setup
 setup(
     name="tigramite",
-    version="5.2.3.1",
+    version="5.2.3.2",
     packages=["tigramite", "tigramite.independence_tests", "tigramite.toymodels"],
     license="GNU General Public License v3.0",
     description="Tigramite causal inference for time series",

diff --git a/tigramite/plotting.py b/tigramite/plotting.py
@@ -450,9 +450,9 @@ class setup_matrix:
     figsize : tuple of floats, optional (default: None)
         Figure size if new figure is created. If None, default pyplot figsize
         is used.
-    minimum : int, optional (default: -1)
+    minimum : float, optional (default: -1.)
         Lower y-axis limit.
-    maximum : int, optional (default: 1)
+    maximum : float, optional (default: 1.)
         Upper y-axis limit.
     label_space_left : float, optional (default: 0.1)
         Fraction of horizontal figure space to allocate left of plot for labels.
@@ -4520,10 +4520,14 @@ def lin_f(x): return x
     #     verbosity=1)
 
 
-    # correlations = pcmci.get_lagged_dependencies(tau_max=20, val_only=True)['val_matrix']
-    # lag_func_matrix = plot_lagfuncs(val_matrix=correlations, setup_args={'label_space_left':0.05, 
-    #                                 'x_base':5, 'y_base':.5})
-    # plt.show()
+    correlations = np.random.rand(3, 3, 5) - 0.5 #pcmci.get_lagged_dependencies(tau_max=20, val_only=True)['val_matrix']
+    lag_func_matrix = plot_lagfuncs(val_matrix=correlations, setup_args={
+                        'label_space_left':0.05, 
+                                    'minimum': 0.0,
+                                    'maximum':.05,
+                                    'x_base':5, 
+                                    'y_base':.5})
+    plt.show()
 
 
     # N = len(links)
@@ -4559,36 +4563,36 @@ def lin_f(x): return x
 
     # val_matrix = np.zeros((4, 4, 3))
 
-    # Complete test case
-    graph = np.zeros((3,3,2), dtype='<U3')
-    val_matrix = 0.*np.random.rand(*graph.shape)
-    val_matrix[:,:,0] = 0.2
-    graph[:] = ""
-    # graph[0, 1, 0] = "<-+"
-    # graph[1, 0, 0] = "+->"
-    graph[0, 0, 1] = "-->"
-    graph[1, 1, 1] = "-->"
+    # # Complete test case
+    # graph = np.zeros((3,3,2), dtype='<U3')
+    # val_matrix = 0.*np.random.rand(*graph.shape)
+    # val_matrix[:,:,0] = 0.2
+    # graph[:] = ""
+    # # graph[0, 1, 0] = "<-+"
+    # # graph[1, 0, 0] = "+->"
+    # graph[0, 0, 1] = "-->"
+    # graph[1, 1, 1] = "-->"
 
-    graph[0, 1, 1] = "+->"
-    # graph[1, 0, 1] = "o-o"
+    # graph[0, 1, 1] = "+->"
+    # # graph[1, 0, 1] = "o-o"
 
     # graph[1, 2, 0] = "<->"
     # graph[2, 1, 0] = "<->"
 
     # graph[0, 2, 0] = "x-x"
-    # graph[2, 0, 0] = "x-x"
-    nolinks = np.zeros(graph.shape)
-    # nolinks[range(4), range(4), 1] = 1
+    # # graph[2, 0, 0] = "x-x"
+    # nolinks = np.zeros(graph.shape)
+    # # nolinks[range(4), range(4), 1] = 1
 
-    # graph = graph[:2, :2, :]
+    # # graph = graph[:2, :2, :]
 
-    # fig, axes = pyplot.subplots(nrows=1, ncols=1, figsize=(6, 5))
+    # # fig, axes = pyplot.subplots(nrows=1, ncols=1, figsize=(6, 5))
 
 
-    # import cartopy.crs as ccrs
-    graph = np.ones((5, 5, 2), dtype='<U3')
-    graph[:] = ""
-    graph[3, :, 1] = '+->' 
+    # # import cartopy.crs as ccrs
+    # graph = np.ones((5, 5, 2), dtype='<U3')
+    # graph[:] = ""
+    # graph[3, :, 1] = '+->' 
 
     # fig = pyplot.figure(figsize=(8, 6))
     # fig = pyplot.figure(figsize=(10, 5))
@@ -4605,18 +4609,18 @@ def lin_f(x): return x
     #         'transform':ccrs.PlateCarree(), # t.PlateCarree()
     #         }
 
-    plot_time_series_graph(graph=graph,
-        # fig_ax = (fig, ax),
-        # val_matrix=val_matrix,
-        # figsize=(5, 5),
-        # var_names = ['Var %s' %i for i in range(len(graph))],
-        # arrow_linewidth=6,
-        # label_space_left = label_space_left,
-        # label_space_top = label_space_top,
-        # # network_lower_bound=network_lower_bound,
-        save_name="tsg_test.pdf"
-        )
-    pyplot.tight_layout()
+    # plot_time_series_graph(graph=graph,
+    #     # fig_ax = (fig, ax),
+    #     # val_matrix=val_matrix,
+    #     # figsize=(5, 5),
+    #     # var_names = ['Var %s' %i for i in range(len(graph))],
+    #     # arrow_linewidth=6,
+    #     # label_space_left = label_space_left,
+    #     # label_space_top = label_space_top,
+    #     # # network_lower_bound=network_lower_bound,
+    #     save_name="tsg_test.pdf"
+    #     )
+    # pyplot.tight_layout()
 
     # network_lower_bound = 0.
     # show_colorbar=True