SQL sensor not finding MySQL package

I wanted to configure this SQL sensor to point to a db on the Raspberry Pi (under the pi user):

  - platform: sql
    db_url: mysql://PI_USER:PI_PASSWORD@PI_IP:DB_PORT/
    scan_interval: 300
    queries:
      - name: QUERY_NAME
        query: 'SELECT * FROM DB_NAME.TABLE_NAME'
        column: 'COLUMN_NAME'

I get this error at component setup:
ModuleNotFoundError: No module named ‘MySQLdb’

I’m on HA 101.3 on a python virtual environment; should I install something in the virtual environment itself?

Take a look in the recorder docs.