Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Test Patch] Remove redundant code for "Fix/update test_run_compressed" #1072

Merged
merged 28 commits into from
Jan 15, 2025

Conversation

horheynm
Copy link
Collaborator

@horheynm horheynm commented Jan 14, 2025

SUMMARY:
Removed breakpoints and addressed comments for #970

TEST PLAN:
Ran pytest for the two test files

#970
ORIGINAL PR DESCRIPTION:
Contingent on merge of huggingface/transformers#34719
^ has been merged not yet released

SUMMARY:
Update run_compressed tests from decompression tests to run_comrpressed tests -> test if run_compressed True/False models generate the same output

Add decompress tests that copies attrs from the source dir path's model to the target model.

TEST PLAN:
ran the test using transformers main
must pass tests/llmcompressor/transformers/compression/test_decompress.py
and tests/llmcompressor/transformers/compression/test_run_compressed.py

@classmethod
def tearDownClass(self):
shutil.rmtree(self.test_dir)
del self.dense_model
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

FYI there's no need to delete references manually. The references are dropped when the reference to the class is dropped, which should be when pytest finishes the test.

@dsikka dsikka merged commit 806da33 into main Jan 15, 2025
6 of 7 checks passed
@dsikka dsikka deleted the run-compressed-test-patch branch January 15, 2025 00:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants