Merge pull request theislab#207 from theislab/fix/esm_test

Reverting the gene embedding test skips

Author: MUCDK

tests/preprocessing/test_gene_emb.py

@@ -52,15 +52,13 @@ def adata_test_legacy():
 
 
 class TestGeneEmb:
-    @pytest.mark.skip(reason="Skipped due to error in transformers package")
     def test_embedding(self, adata_with_ko):
         adata = get_esm_embedding(adata_with_ko, gene_key="gene_target_", copy=True)
         metadata = adata.uns["gene_embedding_metadata"]
         assert Counter(metadata[metadata.is_protein_coding].gene_id.tolist()) == IS_PROT_CODING
         gene_with_prot_seq = metadata[metadata.protein_sequence.notnull()].gene_id.tolist()
         assert Counter(gene_with_prot_seq) == IS_PROT_CODING
 
-    @pytest.mark.skip(reason="Skipped due to error in transformers package")
     def test_legacy_emb(self, adata_test_legacy):
         """Test if we can reproduce the original embeddings we used."""
         adata = get_esm_embedding(adata_test_legacy, gene_key="gene", copy=True)