Skip to content

Commit

Permalink
deploy: 9e16a29
Browse files Browse the repository at this point in the history
  • Loading branch information
@github-actions
github-actions[bot] committed Dec 15, 2023
1 parent 6f46bb4 commit 783ee12
Show file tree
Hide file tree
Showing 39 changed files with 1,514 additions and 573 deletions.
1 change: 1 addition & 0 deletions api/_modules/aepsych/config.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -390,6 +390,7 @@ <h1>Source code for aepsych.config</h1><div class="highlight"><pre>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">gpytorch</span><span class="o">.</span><span class="n">likelihoods</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">gpytorch</span><span class="o">.</span><span class="n">kernels</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">botorch</span><span class="o">.</span><span class="n">acquisition</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">botorch</span><span class="o">.</span><span class="n">acquisition</span><span class="o">.</span><span class="n">multi_objective</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">registered_names</span><span class="p">[</span><span class="s2">"None"</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
</pre></div>
</div>
Expand Down
1 change: 1 addition & 0 deletions api/_modules/aepsych/config/index.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -390,6 +390,7 @@ <h1>Source code for aepsych.config</h1><div class="highlight"><pre>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">gpytorch</span><span class="o">.</span><span class="n">likelihoods</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">gpytorch</span><span class="o">.</span><span class="n">kernels</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">botorch</span><span class="o">.</span><span class="n">acquisition</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">register_module</span><span class="p">(</span><span class="n">botorch</span><span class="o">.</span><span class="n">acquisition</span><span class="o">.</span><span class="n">multi_objective</span><span class="p">)</span>
<span class="n">Config</span><span class="o">.</span><span class="n">registered_names</span><span class="p">[</span><span class="s2">"None"</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span>
</pre></div>
</div>
Expand Down
7 changes: 2 additions & 5 deletions api/_modules/aepsych/generators/optimize_acqf_generator.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ <h1>Source code for aepsych.generators.optimize_acqf_generator</h1><div class="h
<span class="kn">from</span> <span class="nn">aepsych.config</span> <span class="kn">import</span> <span class="n">Config</span><span class="p">,</span> <span class="n">ConfigurableMixin</span>
<span class="kn">from</span> <span class="nn">aepsych.generators.base</span> <span class="kn">import</span> <span class="n">AEPsychGenerationStep</span><span class="p">,</span> <span class="n">AEPsychGenerator</span>
<span class="kn">from</span> <span class="nn">aepsych.models.base</span> <span class="kn">import</span> <span class="n">ModelProtocol</span>
<span class="kn">from</span> <span class="nn">aepsych.models.surrogate</span> <span class="kn">import</span> <span class="n">AEPsychSurrogate</span>
<span class="kn">from</span> <span class="nn">ax.models.torch.botorch_modular.surrogate</span> <span class="kn">import</span> <span class="n">Surrogate</span>
<span class="kn">from</span> <span class="nn">aepsych.utils_logging</span> <span class="kn">import</span> <span class="n">getLogger</span>
<span class="kn">from</span> <span class="nn">ax.modelbridge</span> <span class="kn">import</span> <span class="n">Models</span>
<span class="kn">from</span> <span class="nn">ax.modelbridge.registry</span> <span class="kn">import</span> <span class="n">Cont_X_trans</span>
Expand Down Expand Up @@ -170,14 +170,11 @@ <h1>Source code for aepsych.generators.optimize_acqf_generator</h1><div class="h
<span class="n">acqf_options</span> <span class="o">=</span> <span class="bp">cls</span><span class="o">.</span><span class="n">_get_acqf_options</span><span class="p">(</span><span class="n">acqf_cls</span><span class="p">,</span> <span class="n">config</span><span class="p">)</span>
<span class="n">gen_options</span> <span class="o">=</span> <span class="bp">cls</span><span class="o">.</span><span class="n">_get_gen_options</span><span class="p">(</span><span class="n">config</span><span class="p">)</span>

<span class="n">max_fit_time</span> <span class="o">=</span> <span class="n">model_options</span><span class="p">[</span><span class="s2">"max_fit_time"</span><span class="p">]</span>

<span class="n">model_kwargs</span> <span class="o">=</span> <span class="p">{</span>
<span class="s2">"surrogate"</span><span class="p">:</span> <span class="n">AEPsychSurrogate</span><span class="p">(</span>
<span class="s2">"surrogate"</span><span class="p">:</span> <span class="n">Surrogate</span><span class="p">(</span>
<span class="n">botorch_model_class</span><span class="o">=</span><span class="n">model_class</span><span class="p">,</span>
<span class="n">mll_class</span><span class="o">=</span><span class="n">model_class</span><span class="o">.</span><span class="n">get_mll_class</span><span class="p">(),</span>
<span class="n">model_options</span><span class="o">=</span><span class="n">model_options</span><span class="p">,</span>
<span class="n">max_fit_time</span><span class="o">=</span><span class="n">max_fit_time</span><span class="p">,</span>
<span class="p">),</span>
<span class="s2">"acquisition_class"</span><span class="p">:</span> <span class="n">AEPsychAcquisition</span><span class="p">,</span>
<span class="s2">"botorch_acqf_class"</span><span class="p">:</span> <span class="n">acqf_cls</span><span class="p">,</span>
Expand Down
7 changes: 2 additions & 5 deletions api/_modules/aepsych/generators/optimize_acqf_generator/index.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ <h1>Source code for aepsych.generators.optimize_acqf_generator</h1><div class="h
<span class="kn">from</span> <span class="nn">aepsych.config</span> <span class="kn">import</span> <span class="n">Config</span><span class="p">,</span> <span class="n">ConfigurableMixin</span>
<span class="kn">from</span> <span class="nn">aepsych.generators.base</span> <span class="kn">import</span> <span class="n">AEPsychGenerationStep</span><span class="p">,</span> <span class="n">AEPsychGenerator</span>
<span class="kn">from</span> <span class="nn">aepsych.models.base</span> <span class="kn">import</span> <span class="n">ModelProtocol</span>
<span class="kn">from</span> <span class="nn">aepsych.models.surrogate</span> <span class="kn">import</span> <span class="n">AEPsychSurrogate</span>
<span class="kn">from</span> <span class="nn">ax.models.torch.botorch_modular.surrogate</span> <span class="kn">import</span> <span class="n">Surrogate</span>
<span class="kn">from</span> <span class="nn">aepsych.utils_logging</span> <span class="kn">import</span> <span class="n">getLogger</span>
<span class="kn">from</span> <span class="nn">ax.modelbridge</span> <span class="kn">import</span> <span class="n">Models</span>
<span class="kn">from</span> <span class="nn">ax.modelbridge.registry</span> <span class="kn">import</span> <span class="n">Cont_X_trans</span>
Expand Down Expand Up @@ -170,14 +170,11 @@ <h1>Source code for aepsych.generators.optimize_acqf_generator</h1><div class="h
<span class="n">acqf_options</span> <span class="o">=</span> <span class="bp">cls</span><span class="o">.</span><span class="n">_get_acqf_options</span><span class="p">(</span><span class="n">acqf_cls</span><span class="p">,</span> <span class="n">config</span><span class="p">)</span>
<span class="n">gen_options</span> <span class="o">=</span> <span class="bp">cls</span><span class="o">.</span><span class="n">_get_gen_options</span><span class="p">(</span><span class="n">config</span><span class="p">)</span>

<span class="n">max_fit_time</span> <span class="o">=</span> <span class="n">model_options</span><span class="p">[</span><span class="s2">"max_fit_time"</span><span class="p">]</span>

<span class="n">model_kwargs</span> <span class="o">=</span> <span class="p">{</span>
<span class="s2">"surrogate"</span><span class="p">:</span> <span class="n">AEPsychSurrogate</span><span class="p">(</span>
<span class="s2">"surrogate"</span><span class="p">:</span> <span class="n">Surrogate</span><span class="p">(</span>
<span class="n">botorch_model_class</span><span class="o">=</span><span class="n">model_class</span><span class="p">,</span>
<span class="n">mll_class</span><span class="o">=</span><span class="n">model_class</span><span class="o">.</span><span class="n">get_mll_class</span><span class="p">(),</span>
<span class="n">model_options</span><span class="o">=</span><span class="n">model_options</span><span class="p">,</span>
<span class="n">max_fit_time</span><span class="o">=</span><span class="n">max_fit_time</span><span class="p">,</span>
<span class="p">),</span>
<span class="s2">"acquisition_class"</span><span class="p">:</span> <span class="n">AEPsychAcquisition</span><span class="p">,</span>
<span class="s2">"botorch_acqf_class"</span><span class="p">:</span> <span class="n">acqf_cls</span><span class="p">,</span>
Expand Down
Loading

0 comments on commit 783ee12

Please sign in to comment.