update doctest for Location->Country

SpareCores · Jan 2, 2024 · 7efdf8f · 7efdf8f
1 parent a7024a4
commit 7efdf8f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sc_crawler/schemas.py b/sc_crawler/schemas.py
@@ -51,11 +51,12 @@ class Vendor(SQLModel, table=True):
     """Base class for cloud compute resource vendors.
 
     Examples:
-        >>> from sc_crawler import Location, Vendor, vendors
-        >>> aws_loc = Location(country='US', city='Seattle', address_line1='410 Terry Ave N')
-        >>> aws = Vendor(identifier='aws', name='Amazon Web Services', homepage='https://aws.amazon.com', location=aws_loc, founding_year=2002)
+        >>> from sc_crawler.schemas import Vendor
+        >>> from sc_crawler.lookup import countries
+        >>> aws = Vendor(id='aws', name='Amazon Web Services', homepage='https://aws.amazon.com', country=countries["US"], founding_year=2002)
         >>> aws
         Vendor(identifier='aws'...
+        >>> from sc_crawler import vendors
         >>> vendors.aws
         Vendor(identifier='aws'...
     """  # noqa: E501