yash-b18 commited on
Commit
cce84e6
·
1 Parent(s): 3a73daa

small code fix in retrieving images

Browse files
Files changed (1) hide show
  1. met_api.py +4 -4
met_api.py CHANGED
@@ -18,18 +18,18 @@ def get_objectsWithImages():
18
  objectIDs = response["objectIDs"]
19
  return total, objectIDs
20
 
21
- # gets the urls for 3 random objects with images
22
  def get_images(totalObjects, objectIDs, limit):
23
  images = []
24
  # grabbing extra in case a primary image is blank (Works best on small limits)
25
  rand_indexes = random.sample(range(totalObjects), limit + 20)
26
-
27
  for i in rand_indexes:
28
  obj = get_object(objectIDs[i])
29
  if obj["primaryImage"]:
30
  images.append((obj["primaryImage"], obj["title"]))
31
- limit += 1
32
- if limit == 3:
33
  break
34
 
35
  return images
 
18
  objectIDs = response["objectIDs"]
19
  return total, objectIDs
20
 
21
+ # gets the urls for random objects with images
22
  def get_images(totalObjects, objectIDs, limit):
23
  images = []
24
  # grabbing extra in case a primary image is blank (Works best on small limits)
25
  rand_indexes = random.sample(range(totalObjects), limit + 20)
26
+ counter = 0
27
  for i in rand_indexes:
28
  obj = get_object(objectIDs[i])
29
  if obj["primaryImage"]:
30
  images.append((obj["primaryImage"], obj["title"]))
31
+ counter += 1
32
+ if counter == limit:
33
  break
34
 
35
  return images