Set np printoptions (apache#32539)

reeba212 · Dec 4, 2024 · 86d6214 · 86d6214
1 parent bff7a47
commit 86d6214
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/dataframe/doctests.py b/sdks/python/apache_beam/dataframe/doctests.py
@@ -665,7 +665,10 @@ def set_pandas_options():
   # See
   # https://github.com/pandas-dev/pandas/blob/a00202d12d399662b8045a8dd3fdac04f18e1e55/doc/source/conf.py#L319
   np.random.seed(123456)
-  np.set_printoptions(precision=4, suppress=True)
+  legacy = None
+  if np.version.version.startswith('2'):
+    legacy = '1.25'
+  np.set_printoptions(precision=4, suppress=True, legacy=legacy)
   pd.options.display.max_rows = 15