From 1ba4113405ce8aed090e8c6a8a88b888c8ecdaef Mon Sep 17 00:00:00 2001
From: Huanyu Zhang <huanyuzhang@meta.com>
Date: Sat, 10 Feb 2024 12:15:52 -0800
Subject: [PATCH] release v1.4.1 (#628)

Summary:
Pull Request resolved: https://github.com/pytorch/opacus/pull/628

Releasing v1.4.1

Differential Revision: D53626083

fbshipit-source-id: 32bf2e7b9ffb9544ed6dab8929b4bb901b3967e5
---
 CHANGELOG.md      | 7 +++++++
 conftest.py       | 6 +++++-
 opacus/version.py | 2 +-
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 487cbd77..ed217478 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,12 @@
 # Changelog
 
+## v1.4.1
+
+### Bug fixes
+* Fix DP MultiheadAttention (#598)
+* Fix: make prv accountant robust to larger epsilons (#606)
+* Fix the corner case when the optimizer has no trainable parameters (#619)
+
 ## v1.4
 
 Highlight: Upgraded to PyTorch 1.13+ as required dependency
diff --git a/conftest.py b/conftest.py
index 72898402..c88ec5ee 100644
--- a/conftest.py
+++ b/conftest.py
@@ -60,8 +60,12 @@ def create_namespace(doctest_namespace):
     Initialize namespace for doctest.
     Everything added to `doctest_namespace` will be available in the doctest.
     """
-    from typing import Set, Union  # noqa
+    from typing import Any, Dict, List, Set, Tuple, Union  # noqa
+
+    import numpy as np  # noqa
     import opacus  # noqa
+    import torch  # noqa
+    from torch import nn  # noqa
 
     # Adding all imports in the doctest namespace
     doctest_namespace.update(**locals())
diff --git a/opacus/version.py b/opacus/version.py
index bd041cef..5554678c 100644
--- a/opacus/version.py
+++ b/opacus/version.py
@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.4.0"
+__version__ = "1.4.1"