From bc8595002db32aa2c825104abb796d80387364ee Mon Sep 17 00:00:00 2001
From: Alejandro-Morales <alejandro.morales.mad@gmail.com>
Date: Thu, 22 Jun 2023 09:58:36 -0600
Subject: [PATCH] run pylint

---
 src/uagents/resolver.py          | 10 +++++-----
 tests/test_agent_registration.py |  3 ++-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/uagents/resolver.py b/src/uagents/resolver.py
index 1318ed84..cb1791ec 100644
--- a/src/uagents/resolver.py
+++ b/src/uagents/resolver.py
@@ -57,8 +57,8 @@ async def resolve(self, destination: str) -> Optional[str]:
 
 
 class AlmanacResolver(Resolver):
-    async def resolve(self, address: str) -> Optional[str]:
-        result = query_record(address, "service")
+    async def resolve(self, destination: str) -> Optional[str]:
+        result = query_record(destination, "service")
         if result is not None:
             record = result.get("record") or {}
             endpoint_list = (
@@ -68,14 +68,14 @@ async def resolve(self, address: str) -> Optional[str]:
             if len(endpoint_list) > 0:
                 endpoints = [val.get("url") for val in endpoint_list]
                 weights = [val.get("weight") for val in endpoint_list]
-                return address, random.choices(endpoints, weights=weights)[0]
+                return destination, random.choices(endpoints, weights=weights)[0]
 
         return None, None
 
 
 class NameServiceResolver(Resolver):
-    async def resolve(self, name: str) -> Optional[str]:
-        return get_agent_address(name)
+    async def resolve(self, destination: str) -> Optional[str]:
+        return get_agent_address(destination)
 
 
 class RulesBasedResolver(Resolver):
diff --git a/tests/test_agent_registration.py b/tests/test_agent_registration.py
index 8226616a..70c36f14 100644
--- a/tests/test_agent_registration.py
+++ b/tests/test_agent_registration.py
@@ -3,6 +3,7 @@
 
 from uagents import Agent
 from uagents.setup import fund_agent_if_low
+from uagents.resolver import get_agent_address
 from uagents.config import (
     REGISTRATION_FEE,
     REGISTRATION_DENOM,
@@ -129,7 +130,7 @@ def test_registration(self):
         )
         self.assertEqual(is_owner, True, "Domain ownership failed")
 
-        query_address = agent.get_agent_address(agent.name)
+        query_address = get_agent_address(agent.name)
 
         self.assertEqual(
             query_address == agent.address, True, "Service contract registration failed"