diff --git a/adastop/cli.py b/adastop/cli.py index 52f1ac2..ecc6ac6 100644 --- a/adastop/cli.py +++ b/adastop/cli.py @@ -84,7 +84,7 @@ def compare(ctx, input_file, n_groups, size_group, n_permutations, alpha, seed, if not("continue" in list(comparator.decisions.values())): click.echo('') click.echo("Test is finished, decisions are") - click.echo(comparator.get_results().to_markdown()) + click.echo(comparator.get_results().to_markdown(index=False)) else: still_here = [] diff --git a/tests/test_compare_agents.py b/tests/test_compare_agents.py index 29fe0fd..2a2e51e 100644 --- a/tests/test_compare_agents.py +++ b/tests/test_compare_agents.py @@ -39,7 +39,7 @@ def test_type1(K,n): evals = {"Agent "+str(k): rng.normal(size=n) for k in range(n_agents)} comparator.partial_compare(evals) idxs.append(not("equal" in comparator.decisions.values())) - print(comparator.get_results()) + print(comparator.get_results().to_string(index=False)) assert np.mean(idxs) < 2*alpha + 1/4/(np.sqrt(n_runs)), "type 1 error seems to be too large."