Skip to content

Conversation

@mgarrard
Copy link
Contributor

Summary:
Add warning filters in AxTestCase.setUp to suppress two deprecation warnings that are cluttering test output:

  1. AxClient deprecation warning - The AxClient class is deprecated and will be removed in Ax 1.4.0. Tests using AxClient should not emit these warnings since we're aware of the deprecation.

  2. GenerationStrategy steps deprecation warning - Specifying steps input is no longer supported and will be removed early 2026. Tests using the old API pattern should not emit these warnings.

These warnings are expected in tests and don't provide actionable information during test runs. Suppressing them keeps test output clean and focused on actual issues.

Differential Revision: D91193852

@meta-cla meta-cla bot added the CLA Signed Do not delete this pull request or issue due to inactivity. label Jan 22, 2026
@meta-codesync
Copy link

meta-codesync bot commented Jan 22, 2026

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

@codecov-commenter
Copy link

codecov-commenter commented Jan 22, 2026

Codecov Report

❌ Patch coverage is 96.22642% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.70%. Comparing base (6045c31) to head (841829e).

Files with missing lines Patch % Lines
ax/adapter/cross_validation.py 50.00% 2 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #4800      +/-   ##
==========================================
- Coverage   96.70%   96.70%   -0.01%     
==========================================
  Files         587      587              
  Lines       61294    61306      +12     
==========================================
+ Hits        59275    59285      +10     
- Misses       2019     2021       +2     

☔ 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.

Summary:

I'm not sure why these are showing up as failures today, we migrated to Python scientific stack in july 2024, but these are deprecation warnings we should go ahead and take care of anyway. Figured would knock them out while they are causing me issues

Prompted after some failures in exports not related to my changes: https://github.com/facebook/Ax/actions/runs/21225324302/job/61070638075?fbclid=IwY2xjawPeXMFleHRuA2FlbQIxMQBicmlkETFRTkR6WlE4NHVrd3IyQXNlc3J0YwZhcHBfaWQBMAABHjTAiZi71n24w95hvzEewrKNPKOGzJisgR7t4qJ3APRMYlusgFC-gu7RLiSb_aem_Zk3pmTDonCFsJvZCTkpeMA


Add `.copy()` after `.numpy()` calls to ensure arrays are writeable.
PyTorch tensors converted via `.detach().cpu().numpy()` return read-only
arrays in some cases. The subsequent squeeze operations create read-only
views, and the in-place assignment `loo_covs[:, diag_idx, diag_idx] = loo_vars`
fails with "assignment destination is read-only" error.

Differential Revision: D91185467
…ok#4795)

Summary:

Prompted after some failures in exports not related to my changes: https://github.com/facebook/Ax/actions/runs/21225324302/job/61070638075?fbclid=IwY2xjawPeXMFleHRuA2FlbQIxMQBicmlkETFRTkR6WlE4NHVrd3IyQXNlc3J0YwZhcHBfaWQBMAABHjTAiZi71n24w95hvzEewrKNPKOGzJisgR7t4qJ3APRMYlusgFC-gu7RLiSb_aem_Zk3pmTDonCFsJvZCTkpeMA

Pandas 2.0+ changed default string column dtype from `object` to
`StringDtype(na_value=nan)`. The `_safecast_df()` method doesn't properly
handle the comparison between `StringDtype` and `np.dtype("O")` because
they are different types that don't compare equal.

Add explicit check for `pd.StringDtype` to force casting when needed.

Differential Revision: D91185469
Summary:

Prompted after some failures in exports not related to my changes: https://github.com/facebook/Ax/actions/runs/21225324302/job/61070638075?fbclid=IwY2xjawPeXMFleHRuA2FlbQIxMQBicmlkETFRTkR6WlE4NHVrd3IyQXNlc3J0YwZhcHBfaWQBMAABHjTAiZi71n24w95hvzEewrKNPKOGzJisgR7t4qJ3APRMYlusgFC-gu7RLiSb_aem_Zk3pmTDonCFsJvZCTkpeMA

The `trial_indices` property returns `set[np.int64]` from `.unique()` on
numpy-backed dataframes, but `experiment.trials.keys()` returns Python `int`.
Set operations fail because `np.int64` and `int` are treated as different
types in set comparisons.

Convert numpy types to Python int using set comprehension.

Differential Revision: D91185468
Summary:

The `copy=False` parameter in `DataFrame.astype()` and `DataFrame.reindex()` was deprecated in pandas 2.0 and removed in pandas 3.0. These calls now raise warnings or errors.

Remove the deprecated `copy=False` argument from:
- `arm_data.astype(dtype=column_to_type, copy=False)` in `cast.py`
- `df.reindex(columns=desired_order, copy=False)` in `data.py`

This is part of the ongoing pandas 2.0+ compatibility fixes.

Differential Revision: D91186870
Summary:

Converting a list of numpy arrays directly to a PyTorch tensor is extremely slow because PyTorch iterates through each array individually. This fix converts the list to a single contiguous numpy array first using `np.array()`, which PyTorch can then convert efficiently in one operation.

Performance benchmark (10 constraints × 1000 arms, 100 iterations):
- Before: 1.546 ms per call
- After: 0.428 ms per call
- **Speedup: 3.6x faster**

This eliminates the UserWarning that was cluttering test output and logs.

Differential Revision: D91187902
…k#4799)

Summary:

Pandas 3.0 introduced Copy-on-Write (CoW) which makes chained assignment with `inplace=True` ineffective. The pattern `df["col"].replace(..., inplace=True)` no longer modifies the original DataFrame because the column selection creates a copy.

Replace the chained `inplace=True` pattern with explicit assignment:
- `df["col"].replace(..., inplace=True)` → `df["col"] = df["col"].replace(...)`

This eliminates the `ChainedAssignmentError` warning and ensures the DataFrame is actually modified as intended.

Differential Revision: D91187974
Differential Revision: D91189850
Summary:
The torchx `Runner.stop()` method is deprecated in favor of `cancel()`, which provides identical functionality but is consistent with the CLI and scheduler API naming conventions.

This change updates TorchXRunner to use the non-deprecated method, eliminating PendingDeprecationWarning messages during test runs.

Differential Revision: D91190296
Summary:
Scipy's `pearsonr`, `spearmanr`, and `kendalltau` raise `ConstantInputWarning` when computing correlations on constant arrays, which is expected in model fit diagnostics when predictions or observations have no variance.

This change introduces a context manager to suppress both numpy invalid value warnings and scipy `ConstantInputWarning` in correlation functions. The return type annotation uses `Iterator[None]` which is the correct type for `contextlib.contextmanager` decorated functions.

Differential Revision: D91193099
Summary:

Add warning filters in `AxTestCase.setUp` to suppress two deprecation warnings that are cluttering test output:

1. **AxClient deprecation warning** - The `AxClient` class is deprecated and will be removed in Ax 1.4.0. Tests using AxClient should not emit these warnings since we're aware of the deprecation.

2. **GenerationStrategy steps deprecation warning** - Specifying `steps` input is no longer supported and will be removed early 2026. Tests using the old API pattern should not emit these warnings.

These warnings are expected in tests and don't provide actionable information during test runs. Suppressing them keeps test output clean and focused on actual issues.

Differential Revision: D91193852
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 meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants