#1752 - Removed unnecessary print statement

- Changed docstring on function check_domain_in_cache
This commit is contained in:
SunilSamra
2023-08-21 16:09:17 +01:00
parent 1a13af2f5e
commit 550b62f75d
2 changed files with 6 additions and 3 deletions

View File

@@ -43,8 +43,12 @@ class DNSClient(BaseModel):
"""
self.dns_cache = {}
def check_domain_is_in_cache(self, target_domain: str, session_id: str):
"""Function to check if domain name is in DNS client cache."""
def check_domain_in_cache(self, target_domain: str, session_id: str):
"""Function to check if domain name is in DNS client cache.
:param target_domain: The domain requested for an IP address.
:param session_id: The ID of the session in order to send the response to the DNS server or application.
"""
if target_domain in self.dns_cache:
ip_address = self.dns_cache[target_domain]
self.send(ip_address, session_id)

View File

@@ -45,7 +45,6 @@ def test_seeded_learning(temp_primaite_session):
), "Expected output is based upon a agent that was trained with seed 67890"
session.learn()
actual_mean_reward_per_episode = session.learn_av_reward_per_episode_dict()
print(actual_mean_reward_per_episode, "THISt")
assert actual_mean_reward_per_episode == expected_mean_reward_per_episode