MilesCranmer commited on
Commit
de4d559
1 Parent(s): 82b18ca

Add docstrings to latex export functions

Browse files
Files changed (1) hide show
  1. pysr/export_latex.py +2 -0
pysr/export_latex.py CHANGED
@@ -61,6 +61,7 @@ def generate_single_table(
61
  precision: int = 3,
62
  columns=["equation", "complexity", "loss", "score"],
63
  ):
 
64
  assert isinstance(equations, pd.DataFrame)
65
 
66
  latex_top, latex_bottom = generate_table_environment(columns)
@@ -112,6 +113,7 @@ def generate_multiple_tables(
112
  precision: int = 3,
113
  columns=["equation", "complexity", "loss", "score"],
114
  ):
 
115
 
116
  latex_tables = [
117
  generate_single_table(
 
61
  precision: int = 3,
62
  columns=["equation", "complexity", "loss", "score"],
63
  ):
64
+ """Generate a booktabs-style LaTeX table for a single set of equations."""
65
  assert isinstance(equations, pd.DataFrame)
66
 
67
  latex_top, latex_bottom = generate_table_environment(columns)
 
113
  precision: int = 3,
114
  columns=["equation", "complexity", "loss", "score"],
115
  ):
116
+ """Generate multiple latex tables for a list of equation sets."""
117
 
118
  latex_tables = [
119
  generate_single_table(