How do you draw a sprite from an atlas on to a mesh?

I figured it out. I get the original UVs for the mesh, set the material texture to the atlas, and change the UVs to draw that specific sprite

     public void Init() {
         _originalUVs = _meshFilter.mesh.uv;
         _originalUVsLength = _originalUVs.Length;
     }
 
     public void SetSprite(Sprite sprite) {
         // get the rect the sprite uses on the atlas
         var spriteRect = sprite.rect;
 
          // convert to UV space (between 0 and 1)
         var x = spriteRect.x / sprite.texture.width;
         var y = spriteRect.y / sprite.texture.height;
         var w = spriteRect.width / sprite.texture.width;
         var h = spriteRect.height / sprite.texture.height;
 
          // set uvs for this sprite
         var newUVs = new Vector2[_originalUVsLength];
         for(var i = 0; i < _originalUVsLength; i++) {
             newUVs[i] = new Vector2(x + (w * _originalUVs[i].x), y + (h * _originalUVs[i].y));
         }
 
         _meshFilter.mesh.uv = newUVs;
         _material.mainTexture = sprite.texture;
     }