Skip to content

Don't override explicitly requested TL model class (#5183)#5183

Closed
hvarfner wants to merge 1 commit intofacebook:mainfrom
hvarfner:export-D101174566
Closed

Don't override explicitly requested TL model class (#5183)#5183
hvarfner wants to merge 1 commit intofacebook:mainfrom
hvarfner:export-D101174566

Conversation

@hvarfner
Copy link
Copy Markdown

@hvarfner hvarfner commented Apr 21, 2026

Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to HeterogeneousMTGP. This prevents using other models at all.

Now, we only override to HeterogeneousMTGP in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566

@meta-cla meta-cla Bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Apr 21, 2026
@meta-codesync
Copy link
Copy Markdown

meta-codesync Bot commented Apr 21, 2026

@hvarfner has exported this pull request. If you are a Meta employee, you can view the originating Diff in D101174566.

hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 21, 2026
…ing_gs (facebook#5183)

Summary:

When the source and target search spaces are incompatible,
`get_transfer_learning_gs` silently overrides the model class to
`HeterogeneousMTGP`. This prevents using `ImputedMultiTaskGP` or
other heterogeneous-aware model classes that the caller explicitly
requested.

Only override to `HeterogeneousMTGP` when the caller passed the
default `MultiTaskGP`. If a different model class was explicitly
requested, respect it.

Differential Revision: D101174566
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 21, 2026
…ing_gs (facebook#5183)

Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Differential Revision: D101174566
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 21, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Differential Revision: D101174566
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 21, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Differential Revision: D101174566
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 21, 2026
…ing_gs (facebook#5183)

Summary:
Pull Request resolved: facebook#5183

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Differential Revision: D101174566
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 22, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@meta-codesync meta-codesync Bot changed the title Don't override explicitly requested model class in get_transfer_learning_gs Don't override explicitly requested TL model class (#5183) Apr 22, 2026
@hvarfner hvarfner force-pushed the export-D101174566 branch from aaa839d to 5a8da98 Compare April 22, 2026 15:59
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 23, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 5a8da98 to 9f13a2f Compare April 23, 2026 05:53
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 23, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 9f13a2f to 31aedea Compare April 23, 2026 16:00
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 23, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 31aedea to 201b91e Compare April 23, 2026 19:27
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 23, 2026
Summary:
Pull Request resolved: facebook#5183

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 201b91e to effb38e Compare April 23, 2026 19:30
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from effb38e to 88a3e29 Compare April 24, 2026 02:59
@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Apr 24, 2026

Codecov Report

❌ Patch coverage is 77.27273% with 10 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.42%. Comparing base (eeb78fe) to head (3c86c96).

Files with missing lines Patch % Lines
ax/generators/torch/botorch_modular/utils.py 64.28% 10 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #5183      +/-   ##
==========================================
- Coverage   96.43%   96.42%   -0.02%     
==========================================
  Files         617      617              
  Lines       69379    69398      +19     
==========================================
+ Hits        66908    66918      +10     
- Misses       2471     2480       +9     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 88a3e29 to 3ca2bf0 Compare April 24, 2026 03:39
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 3ca2bf0 to b2b7a0a Compare April 24, 2026 15:40
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:
Pull Request resolved: facebook#5183

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from b2b7a0a to d852e7a Compare April 24, 2026 15:45
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from d852e7a to 3813b94 Compare April 24, 2026 16:08
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 3813b94 to 81e0ce3 Compare April 24, 2026 16:09
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from 81e0ce3 to e4a3800 Compare April 24, 2026 16:11
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:
Pull Request resolved: facebook#5183

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from e4a3800 to e5f9508 Compare April 24, 2026 16:12
hvarfner pushed a commit to hvarfner/Ax that referenced this pull request Apr 24, 2026
Summary:

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@hvarfner hvarfner force-pushed the export-D101174566 branch from e5f9508 to 1172f3a Compare April 24, 2026 16:13
Summary:
Pull Request resolved: facebook#5183

When the source and target search spaces are incompatible,
any model class is silently overridden to `HeterogeneousMTGP`. This prevents using other models at all.

Now, we only  override to `HeterogeneousMTGP` in get_transfer_learning_gs, instead of in two different places in the stack. Thus, we will still have the same functionality but not double-force it.

Reviewed By: sdaulton

Differential Revision: D101174566
@meta-codesync
Copy link
Copy Markdown

meta-codesync Bot commented Apr 24, 2026

This pull request has been merged in ebad6c1.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed Do not delete this pull request or issue due to inactivity. fb-exported Merged meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants