From 69bf6cb3d837cb73e7ac80caf276a06b88ab2899 Mon Sep 17 00:00:00 2001
From: SeunghyunSEO <seosh7039@gmail.com>
Date: Wed, 13 Nov 2024 10:32:11 -0800
Subject: [PATCH] import bugfix (#46)

Summary:
```
  File "/workspace/optimizers/distributed_shampoo/distributed_shampoo.py", line 20, in <module>
    from distributed_shampoo.shampoo_types import (
  File "/workspace/optimizers/distributed_shampoo/shampoo_types.py", line 15, in <module>
    from commons import AbstractDataclass
ModuleNotFoundError: No module named 'commons'
```

after fix

```
python -c "from distributed_shampoo.distributed_shampoo import DistributedShampoo;
shampoo_optimizer = DistributedShampoo;
print(shampoo_optimizer)"
<class 'distributed_shampoo.distributed_shampoo.DistributedShampoo'>
```

Pull Request resolved: https://github.com/facebookresearch/optimizers/pull/46

Reviewed By: anana10c

Differential Revision: D65886425

Pulled By: tsunghsienlee

fbshipit-source-id: e0f4658578dc1c02dfee8f96834e17c2a2ed12e2
---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index cec25be..8faf52b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -52,6 +52,7 @@ Repository = "https://github.com/facebookresearch/optimizers.git"
 [tool.setuptools]
 include-package-data = false
 py-modules = [
+    "commons",
     "matrix_functions",
     "matrix_functions_types",
     "optimizer_modules",