| # Copyright (c) Meta Platforms, Inc. and affiliates. |
| # All rights reserved. |
| # |
| # This source code is licensed under the BSD-style license found in the |
| # LICENSE file in the root directory of this source tree. |
| |
| """ |
| This removes the imported [examples] module from pip installing lm_eval due to |
| colliding module name with ET package |
| """ |
| |
| try: |
| # If the import fails, this means there's nothing to remove |
| import examples |
| |
| try: |
| # If the import succeeds, this means it isn't using lm_eval's module |
| import examples.models |
| except: |
| print( |
| "Failed to import examples.models due to lm_eval conflict. Removing lm_eval examples module" |
| ) |
| import shutil |
| |
| examples_path = examples.__path__[0] |
| shutil.rmtree(examples_path) |
| |
| except: |
| pass |
