use dot get for prefix check

opentensor · Jun 22, 2023 · 9bd4a82 · 9bd4a82
1 parent fedeabe
commit 9bd4a82
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/bittensor/_axon/__init__.py b/bittensor/_axon/__init__.py
@@ -154,7 +154,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None):
         """Accept specific arguments from parser"""
         prefix_str = "" if prefix is None else prefix + "."
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).axon = bittensor.defaults.axon
 

diff --git a/bittensor/_dataset/__init__.py b/bittensor/_dataset/__init__.py
@@ -135,7 +135,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         """
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).dataset = bittensor.defaults.dataset
         try:

diff --git a/bittensor/_logging/__init__.py b/bittensor/_logging/__init__.py
@@ -144,7 +144,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         """
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).logging = bittensor.defaults.logging
         try:

diff --git a/bittensor/_prometheus/__init__.py b/bittensor/_prometheus/__init__.py
@@ -150,7 +150,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         """
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).prometheus = bittensor.defaults.prometheus
         try:

diff --git a/bittensor/_subtensor/__init__.py b/bittensor/_subtensor/__init__.py
@@ -142,7 +142,7 @@ def help(cls):
     def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).subtensor = bittensor.defaults.subtensor
         try:

diff --git a/bittensor/_threadpool/__init__.py b/bittensor/_threadpool/__init__.py
@@ -56,7 +56,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         """
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).priority = bittensor.defaults.priority
         try:

diff --git a/bittensor/_wallet/__init__.py b/bittensor/_wallet/__init__.py
@@ -107,7 +107,7 @@ def add_args(cls, parser: argparse.ArgumentParser, prefix: str = None ):
         """
         prefix_str = '' if prefix == None else prefix + '.'
         if prefix is not None:
-            if not hasattr(bittensor.defaults, prefix):
+            if bittensor.defaults.get(prefix, d=None) == None:
                 setattr(bittensor.defaults, prefix, bittensor.Config())
             getattr(bittensor.defaults, prefix).wallet = bittensor.defaults.wallet
         try: