Skip to content

Commit

Permalink
Merge pull request #138 from kkaris/mti-mesh-dumper-fix
Browse files Browse the repository at this point in the history
Fix attribute misnomer
  • Loading branch information
pagreene committed Oct 8, 2020
2 parents da8ad27 + 9fbee3d commit 00e24fc
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions indra_db/managers/dump_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -175,9 +175,16 @@ class SourceCount(Dumper):
name = 'source_count'
fmt = 'pkl'

def __init__(self, db_label='primary', use_principal=True, **kwargs):
self.use_principal = use_principal
super(SourceCount, self).__init__(db_label, **kwargs)

def dump(self, continuing=False):
db = get_db(self.db_label)
get_source_counts(self.get_s3_path(), db)
if self.use_principal:
ro = get_db(self.db_label)
else:
ro = get_ro(self.db_label)
get_source_counts(self.get_s3_path(), ro)


class FullPaJson(Dumper):
Expand Down Expand Up @@ -238,7 +245,7 @@ def dump(self, continuing=False):

class StatementHashMeshId(Dumper):
name = 'mti_mesh_ids'
fmi = 'pkl'
fmt = 'pkl'

def __init__(self, db_label='primary', use_principal=False, **kwargs):
self.use_principal = use_principal
Expand Down

0 comments on commit 00e24fc

Please sign in to comment.