Merge pull request #133 from bmcfee/bugfix

Fixed a bug in opmaps
bmcfee · Apr 13, 2022 · 057d114 · 057d114
2 parents 0740316 + 4532678
commit 057d114
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,7 +1,7 @@
 # pumpp
 [![GitHub license](https://img.shields.io/badge/license-ISC-blue.svg)](https://raw.githubusercontent.com/bmcfee/pumpp/master/LICENSE)
-[![Build Status](https://travis-ci.org/bmcfee/pumpp.svg?branch=master)](https://travis-ci.org/bmcfee/pumpp)
-[![Coverage Status](https://coveralls.io/repos/github/bmcfee/pumpp/badge.svg?branch=master)](https://coveralls.io/github/bmcfee/pumpp?branch=master)
+[![CI](https://github.com/bmcfee/pumpp/actions/workflows/ci.yml/badge.svg)](https://github.com/bmcfee/pumpp/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/gh/bmcfee/pumpp/branch/main/graph/badge.svg?token=i7YxIwweBQ)](https://codecov.io/gh/bmcfee/pumpp)
 [![Documentation Status](http://readthedocs.org/projects/pumpp/badge/?version=latest)](http://pumpp.readthedocs.io/en/latest/?badge=latest)
 
 

diff --git a/pumpp/core.py b/pumpp/core.py
@@ -226,7 +226,7 @@ def layers(self, api='keras'):
         return layermap
 
     def __getitem__(self, key):
-        return self.opmap.get(key)
+        return self.opmap[key]
 
     def __call__(self, *args, **kwargs):
         return self.transform(*args, **kwargs)