fix fatcat_transform state filters

1 files changed, 4 insertions, 4 deletions

diff --git a/python/fatcat_transform.py b/python/fatcat_transform.py
index 9ddbaa4d..23a56109 100755
--- a/python/fatcat_transform.py
+++ b/python/fatcat_transform.py
@@ -30,7 +30,7 @@ def run_elasticsearch_releases(args):
         if not line:
             continue
         entity = entity_from_json(line, ReleaseEntity, api_client=args.api.api_client)
-        if entity['state'] != 'active':
+        if entity.state != 'active':
             continue
         args.json_output.write(
             json.dumps(release_to_elasticsearch(entity)) + '\n')
@@ -41,7 +41,7 @@ def run_elasticsearch_containers(args):
         if not line:
             continue
         entity = entity_from_json(line, ContainerEntity, api_client=args.api.api_client)
-        if entity['state'] != 'active':
+        if entity.state != 'active':
             continue
         args.json_output.write(
             json.dumps(container_to_elasticsearch(entity)) + '\n')
@@ -52,7 +52,7 @@ def run_elasticsearch_files(args):
         if not line:
             continue
         entity = entity_from_json(line, FileEntity, api_client=args.api.api_client)
-        if entity['state'] != 'active':
+        if entity.state != 'active':
             continue
         args.json_output.write(
             json.dumps(file_to_elasticsearch(entity)) + '\n')
@@ -72,7 +72,7 @@ def run_citeproc_releases(args):
         if not line:
             continue
         entity = entity_from_json(line, ReleaseEntity, api_client=args.api.api_client)
-        if entity['state'] != 'active':
+        if entity.state != 'active':
             continue
         csl_json = release_to_csl(entity)
         csl_json['id'] = "release:" + (entity.ident or "unknown")