Skip to content

Commit

Permalink
Remove n_culled_sample_below everywhere
Browse files Browse the repository at this point in the history
  • Loading branch information
vinhowe committed May 20, 2023
1 parent 7435097 commit c5b8021
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions estimate_survey.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@ def estimate_survey_costs(
*,
n_samples_per_dependent_variable: typing.Optional[int] = None,
n_top_mutual_info_dvs: typing.Optional[int] = None,
n_cull_sampled_below: typing.Optional[int] = None,
):
# TODO(vinhowe): fix this
survey_directory = survey_name
Expand Down Expand Up @@ -65,8 +64,7 @@ def estimate_survey_costs(

dependent_variable_samples = list(
survey.iterate(
n_samples_per_dependent_variable=n_samples_per_dependent_variable,
n_cull_sampled_below=n_cull_sampled_below,
n_samples_per_dependent_variable=n_samples_per_dependent_variable
)
)

Expand Down Expand Up @@ -113,7 +111,6 @@ def main(
survey_names: typing.List[str],
n_samples_per_dependent_variable: typing.Optional[int] = None,
n_top_mutual_info_dvs: typing.Optional[int] = None,
n_cull_sampled_below: typing.Optional[int] = None,
) -> None:
sampler = AutoSampler(model_name=model_name)

Expand All @@ -124,7 +121,6 @@ def main(
survey_name=survey_name,
n_samples_per_dependent_variable=n_samples_per_dependent_variable,
n_top_mutual_info_dvs=n_top_mutual_info_dvs,
n_cull_sampled_below=n_cull_sampled_below,
)
survey_costs[survey_name] = estimate

Expand Down Expand Up @@ -159,10 +155,6 @@ def main(
"--n_samples_per_dependent_variable",
type=int,
)
parser.add_argument(
"--n_cull_sampled_below",
type=int,
)
parser.add_argument(
"--n_top_mutual_info_dvs",
type=int,
Expand All @@ -181,5 +173,4 @@ def main(
survey_names=args.survey_name,
n_samples_per_dependent_variable=args.n_samples_per_dependent_variable,
n_top_mutual_info_dvs=args.n_top_mutual_info_dvs,
n_cull_sampled_below=args.n_cull_sampled_below,
)

0 comments on commit c5b8021

Please sign in to comment.