Search code examples
pythonrasa-core

AttributeError: 'NoneType' object has no attribute 'iterrows'


I am trying to build a chatbot in rasa. But my class ActionSearchRestaurants is throwing this error:

for index, row in resrnt.iterrows(): AttributeError: 'NoneType' object has no attribute 'iterrows'

Here is my class ActionSearchRestaurants that I'm using.

class ActionSearchRestaurants(Action):
    def name(self):
        return 'action_restaurant'

    def run(self, dispatcher, tracker, domain):
        config={ "user_key":"16cde****e0a12d10a7bc8bff6568031"}
        zomato = zomatopy.initialize_app(config)
        loc = tracker.get_slot('location')
        cuisine = tracker.get_slot('cuisine')
        location_detail=zomato.get_location(loc, 1)

        cols = ['restaurant_name', 'restaurant_address', 'avg_budget_for_two', 'zomato_rating']
        resrnt = pd.DataFrame(columns = cols)

        d1 = json.loads(location_detail)
        lat=d1["location_suggestions"][0]["latitude"]
        lon=d1["location_suggestions"][0]["longitude"]
        cuisines_dict={'bakery':5,'chinese':25,'cafe':30,'italian':55,'biryani':7,'north indian':50,'south indian':85}
        results=zomato.restaurant_search("", lat, lon, str(cuisines_dict.get(cuisine)), 10)
        d = json.loads(results)
        response=""
        if d['results_found'] == 0:
            response= "no results"
        else:
            for restaurant in d['restaurants']:
                curr_res = {'zomato_rating':restaurant['restaurant']["user_rating"]["aggregate_rating"],'restaurant_name':restaurant['restaurant']['name'],'restaurant_address': restaurant['restaurant']['location']['address'], 'avg_budget_for_two': restaurant['restaurant']['average_cost_for_two']}
                resrnt = resrnt.append(curr_res, ignore_index=True)

        resrnt=resrnt.sort_values(by=['zomato_rating'], ascending=False, inplace=True)

        for index, row in resrnt.iterrows():
                response = response+ index + ". Found \""+ row['restaurant_name']+ "\" in "+ row['restaurant_address']+" has been rated "+ row['zomato_rating']+"\n"

        dispatcher.utter_message("-----"+ response)
        return [SlotSet('location',loc)]

Solution

  • On this line:

    resrnt=resrnt.sort_values(by=['zomato_rating'], ascending=False, inplace=True)
    

    From the documentation for Dataframe.sort_values:

    Returns: sorted_obj : DataFrame or None

    DataFrame with sorted values if inplace=False, None otherwise.

    Since inplace=True, the DataFrame is replaced with None, which of course does not have a .iterrows.

    Either use inplace=False (or omit it), or do not reassign. (One specific reason for using inplace=True is so as not to need the reassignment.)